diff --git a/visualpic/visualization/vtk_visualizer.py b/visualpic/visualization/vtk_visualizer.py
index 2f8954c..299b911 100644
--- a/visualpic/visualization/vtk_visualizer.py
+++ b/visualpic/visualization/vtk_visualizer.py
@@ -1205,6 +1205,7 @@ def get_colormap(self):
 
     def set_colormap(self, cmap):
         self.cmap = cmap
+        cmap = self.get_colormap()
         self._set_vtk_colormap(cmap)
 
     def get_optimized_opacity(self, time_step, bins=11,