diff --git a/README.md b/README.md
index 6473e1eb3..25b8d1e48 100644
--- a/README.md
+++ b/README.md
@@ -131,7 +131,7 @@ print(lerobot.available_datasets)
 # >>> ['aloha_sim_insertion_human', 'aloha_sim_insertion_scripted', 'aloha_sim_transfer_cube_human', 'aloha_sim_transfer_cube_scripted', 'pusht', 'xarm_lift_medium']
 
 # TODO(rcadene): remove DATA_DIR
-dataset = AlohaDataset("aloha_sim_transfer_cube_human", root=Path(os.environ.get("DATA_DIR")))
+dataset = AlohaDataset("pusht", root=Path(os.environ.get("DATA_DIR")))
 
 video_paths = render_dataset(
     dataset,
@@ -139,14 +139,13 @@ video_paths = render_dataset(
     max_num_episodes=1,
 )
 print(video_paths)
-# ['outputs/visualize_dataset/example/episode_0_top.mp4']
+# ['outputs/visualize_dataset/example/episode_0.mp4']
 ```
 
 Or you can achieve the same result by executing our script from the command line:
 ```bash
 python lerobot/scripts/visualize_dataset.py \
-env=aloha \
-task=sim_sim_transfer_cube_human \
+env=pusht \
 hydra.run.dir=outputs/visualize_dataset/example
 # >>> ['outputs/visualize_dataset/example/episode_0.mp4']
 ```
diff --git a/examples/1_visualize_dataset.py b/examples/1_visualize_dataset.py
index bfc2635f6..e9b62ff12 100644
--- a/examples/1_visualize_dataset.py
+++ b/examples/1_visualize_dataset.py
@@ -9,7 +9,7 @@ print(lerobot.available_datasets)
 # >>> ['aloha_sim_insertion_human', 'aloha_sim_insertion_scripted', 'aloha_sim_transfer_cube_human', 'aloha_sim_transfer_cube_scripted', 'pusht', 'xarm_lift_medium']
 
 # TODO(rcadene): remove DATA_DIR
-dataset = AlohaDataset("aloha_sim_transfer_cube_human", root=Path(os.environ.get("DATA_DIR")))
+dataset = AlohaDataset("pusht", root=Path(os.environ.get("DATA_DIR")))
 
 video_paths = render_dataset(
     dataset,
@@ -17,4 +17,4 @@ video_paths = render_dataset(
     max_num_episodes=1,
 )
 print(video_paths)
-# ['outputs/visualize_dataset/example/episode_0_top.mp4']
+# ['outputs/visualize_dataset/example/episode_0.mp4']