diff --git a/src/robofish/io/file.py b/src/robofish/io/file.py
index b81711a1fff8adfa20346677247752bc1a64e6e6..d6ce9bd2276ab6f7f303e491d24a3548179f1033 100644
--- a/src/robofish/io/file.py
+++ b/src/robofish/io/file.py
@@ -37,7 +37,7 @@ import platform
 
 import matplotlib
 import matplotlib.pyplot as plt
-import matplotlib.cm as cm
+import matplotlib.cm
 from matplotlib import animation
 from matplotlib import patches
 
@@ -928,7 +928,7 @@ class File(h5py.File):
         else:
             step_size = poses.shape[1]
 
-        cmap = cm.get_cmap(cmap)
+        cmap = matplotlib.cm.get_cmap(cmap)
 
         x_world, y_world = self.world_size
         if figsize is None: