diff --git a/lerobot/scripts/visualize_dataset.py b/lerobot/scripts/visualize_dataset.py
index ed95b39a..4b7b7d6c 100644
--- a/lerobot/scripts/visualize_dataset.py
+++ b/lerobot/scripts/visualize_dataset.py
@@ -79,7 +79,7 @@ def render_dataset(dataset, out_dir, max_num_episodes):
 
         out_dir.mkdir(parents=True, exist_ok=True)
         for im_key in dataset.image_keys:
-            if len(dataset.image_keys) > 0:
+            if len(dataset.image_keys) > 1:
                 im_name = im_key.replace("observation.images.", "")
                 video_path = out_dir / f"episode_{ep_id}_{im_name}.mp4"
             else: