diff --git a/.github/workflows/python-test.yml b/.github/workflows/python-test.yml
index ec756667..cf40421b 100644
--- a/.github/workflows/python-test.yml
+++ b/.github/workflows/python-test.yml
@@ -35,5 +35,5 @@ jobs:
       run: |
         pip install torchvision
         cd examples/mnist
-        python train.py --env UNIT
-        python compute_influences.py --env UNIT
+        python train.py
+        python compute_influences.py