diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index e4aa4a30..0ad773f3 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -115,10 +115,6 @@ jobs:
             env=pusht \
             wandb.enable=False \
             eval_episodes=1 \
+            env.episode_length=8 \
             device=cpu \
             policy.pretrained_model_path=tests/outputs/models/1.pt
-      #----------------------------------------------
-      #              cleanup
-      #----------------------------------------------
-      - name: Cleanup
-        run: rm -rf diffusion_policy data