diff --git a/tests/e2e/solo/test_grpo.py b/tests/e2e/solo/test_grpo.py
index 5a1ce2ed8..7ae7094ed 100644
--- a/tests/e2e/solo/test_grpo.py
+++ b/tests/e2e/solo/test_grpo.py
@@ -68,6 +68,7 @@ def oai_gsm8k_transform(cfg, *args, **kwargs):
                         "type": "rewards.oai_gsm8k_transform",
                     },
                 ],
+                "output_dir": temp_dir,
             }
         )