diff --git a/src/pan3d/widgets/color_by.py b/src/pan3d/widgets/color_by.py
index 72b496c..4cba072 100644
--- a/src/pan3d/widgets/color_by.py
+++ b/src/pan3d/widgets/color_by.py
@@ -371,7 +371,7 @@ def set_data_arrays_from_vtk(self, dataset, associations=None):
                 for array in arrays:
                     array_info.append(
                         {
-                            "name": array.GetName(),
+                            "name": array.VTKObject.GetName(),
                             "min": np.min(array),
                             "max": np.max(array),
                             "assoc": association,