diff --git a/src/robofish/io/file.py b/src/robofish/io/file.py
index 08dabb9f1c8845330e6044981a3627f16fe0c48c..e07a252cb4388e562afd749806f6e772fd5e3543 100644
--- a/src/robofish/io/file.py
+++ b/src/robofish/io/file.py
@@ -927,6 +927,8 @@ class File(h5py.File):
                 this_c = cmap(fish_id)
             elif isinstance(c, list):
                 this_c = c[fish_id]
+            else:
+                this_c = c
 
             timesteps = poses.shape[1] - 1
             for t in range(0, timesteps, step_size):