diff --git a/src/ansys/meshing/prime/graphics/graphics.py b/src/ansys/meshing/prime/graphics/graphics.py
index 39471d30cf..2666b302b4 100644
--- a/src/ansys/meshing/prime/graphics/graphics.py
+++ b/src/ansys/meshing/prime/graphics/graphics.py
@@ -484,7 +484,7 @@ def __call__(
         else:
             self.show(update)
 
-    def __color_by_type_callback(self):  # pragma: no cover
+    def __color_by_type_callback(self, button_value: bool):  # pragma: no cover
         """Determine the color of a type in the callback."""
         vr = self._colorByTypeBt.GetRepresentation()
         state = vr.GetState()
@@ -510,7 +510,7 @@ def __color_by_type_callback(self):  # pragma: no cover
             for disp_mesh in data["faces"]
         ]
 
-    def __hide_unhide_selection(self):  # pragma: no cover
+    def __hide_unhide_selection(self, button_value: bool):  # pragma: no cover
         """Hide or unhide the clicked component."""
         sel_disp_mesh = self._picker.selections
         if len(sel_disp_mesh) > 0:
@@ -601,7 +601,6 @@ def show(self, update=False):
         if self._sphinx_build == False:
             self._colorByTypeBt = self._plotter.add_checkbox_button_widget(
                 self.__color_by_type_callback,
-                value=int(self._color_by_type) is ColorByType.ZONELET,
                 position=(10, 700),
                 size=30,
                 border_size=3,
@@ -683,7 +682,6 @@ def __draw_parts(self, parts: List = [], update: bool = False, spline: bool = Fa
         if self._sphinx_build == False:
             self._colorByTypeBt = self._plotter.add_checkbox_button_widget(
                 self.__color_by_type_callback,
-                value=self._color_by_type is ColorByType.ZONELET,
                 position=(10, 700),
                 size=30,
                 border_size=3,
@@ -751,7 +749,6 @@ def __draw_scope_parts(self, update=False, scope: prime.ScopeDefinition = None):
         if self._sphinx_build == False:
             self._colorByTypeBt = self._plotter.add_checkbox_button_widget(
                 self.__color_by_type_callback,
-                value=self._color_by_type is ColorByType.ZONELET,
                 position=(10, 700),
                 size=30,
                 border_size=3,