diff --git a/lerobot/common/envs/factory.py b/lerobot/common/envs/factory.py
index 35ebfa4a..dd8ab2f7 100644
--- a/lerobot/common/envs/factory.py
+++ b/lerobot/common/envs/factory.py
@@ -18,7 +18,7 @@ def make_env(cfg, transform=None):
         kwargs["task"] = cfg.env.task
         clsfunc = SimxarmEnv
     elif cfg.env.name == "pusht":
-        from lerobot.common.envs.pusht.pusht import PushtEnv
+        from lerobot.common.envs.pusht.env import PushtEnv
 
         # assert kwargs["seed"] > 200, "Seed 0-200 are used for the demonstration dataset, so we don't want to seed the eval env with this range."
 
diff --git a/tests/test_envs.py b/tests/test_envs.py
index 48e637d5..b51c441b 100644
--- a/tests/test_envs.py
+++ b/tests/test_envs.py
@@ -3,7 +3,7 @@ from tensordict import TensorDict
 from torchrl.envs.utils import check_env_specs, step_mdp
 
 from lerobot.common.envs.factory import make_env
-from lerobot.common.envs.pusht.pusht import PushtEnv
+from lerobot.common.envs.pusht.env import PushtEnv
 from lerobot.common.envs.simxarm import SimxarmEnv
 
 from .utils import init_config