diff --git a/src/Python/CompositeData/CompositePolyDataMapper.py b/src/Python/CompositeData/CompositePolyDataMapper.py
index 8676506fd152d7fc6e23e535d8d7cb444e1a45d0..9471ab8aec489112d1c905c13b36d7fb0eb9132f 100755
--- a/src/Python/CompositeData/CompositePolyDataMapper.py
+++ b/src/Python/CompositeData/CompositePolyDataMapper.py
@@ -2,17 +2,19 @@
 
 # noinspection PyUnresolvedReferences
 import vtkmodules.vtkInteractionStyle
+# noinspection PyUnresolvedReferences
+import vtkmodules.vtkRenderingOpenGL2
 from vtkmodules.vtkCommonColor import vtkNamedColors
 from vtkmodules.vtkCommonDataModel import vtkMultiBlockDataSet
 from vtkmodules.vtkFiltersSources import vtkSphereSource
 from vtkmodules.vtkRenderingCore import (
     vtkActor,
     vtkCompositeDataDisplayAttributes,
+    vtkCompositePolyDataMapper,
     vtkRenderWindow,
     vtkRenderWindowInteractor,
     vtkRenderer
 )
-from vtkmodules.vtkRenderingOpenGL2 import vtkCompositePolyDataMapper2
 
 
 def main():
@@ -39,7 +41,8 @@ def main():
     # NULL in this process.
     mbds.SetBlock(2, sphere2.GetOutput())
 
-    mapper = vtkCompositePolyDataMapper2()
+    # Use vtkCompositePolyDataMapper2 if VTK Version < 90020230516
+    mapper = vtkCompositePolyDataMapper()
     mapper.SetInputDataObject(mbds)
     cdsa = vtkCompositeDataDisplayAttributes()
     mapper.SetCompositeDataDisplayAttributes(cdsa)