diff --git a/examples/1_visualize_dataset.py b/examples/1_visualize_dataset.py
index ec70ac27..15e0e54d 100644
--- a/examples/1_visualize_dataset.py
+++ b/examples/1_visualize_dataset.py
@@ -2,7 +2,7 @@ import os
 from pathlib import Path
 
 import lerobot
-from lerobot.common.datasets.aloha import PushtDataset
+from lerobot.common.datasets.pusht import PushtDataset
 from lerobot.scripts.visualize_dataset import render_dataset
 
 print(lerobot.available_datasets)