diff --git a/lerobot/__init__.py b/lerobot/__init__.py
index 8e554998..a5a90fb4 100644
--- a/lerobot/__init__.py
+++ b/lerobot/__init__.py
@@ -138,8 +138,8 @@ available_policies = [
 
 # keys and values refer to yaml files
 available_policies_per_env = {
-    "aloha": ["act", "diffusion"],
-    "pusht": ["act", "diffusion"],
+    "aloha": ["act"],
+    "pusht": ["diffusion"],
     "xarm": ["tdmpc"],
     "dora_aloha_real": ["act_real"],
 }