diff --git a/python/damask/_vtk.py b/python/damask/_vtk.py
index 561726153..ab2cc5270 100644
--- a/python/damask/_vtk.py
+++ b/python/damask/_vtk.py
@@ -21,7 +21,7 @@ class VTK:
     High-level interface to VTK.
     """
 
-    def __init__(self, vtk_data: vtk.vtkImageData):
+    def __init__(self, vtk_data: vtk.vtkDataSet):
         """
         New spatial visualization.