diff --git a/tests/policies/test_pi05_openpi.py b/tests/policies/pi0_pi05/test_pi05_openpi.py
similarity index 100%
rename from tests/policies/test_pi05_openpi.py
rename to tests/policies/pi0_pi05/test_pi05_openpi.py
diff --git a/tests/policies/test_pi0_openpi.py b/tests/policies/pi0_pi05/test_pi0_openpi.py
similarity index 100%
rename from tests/policies/test_pi0_openpi.py
rename to tests/policies/pi0_pi05/test_pi0_openpi.py
diff --git a/tests/policies/test_pi0_original_vs_lerobot.py b/tests/policies/pi0_pi05/test_pi0_original_vs_lerobot.py
similarity index 100%
rename from tests/policies/test_pi0_original_vs_lerobot.py
rename to tests/policies/pi0_pi05/test_pi0_original_vs_lerobot.py
diff --git a/tests/policies/test_pi0_pi05_hub.py b/tests/policies/pi0_pi05/test_pi0_pi05_hub.py
similarity index 100%
rename from tests/policies/test_pi0_pi05_hub.py
rename to tests/policies/pi0_pi05/test_pi0_pi05_hub.py