diff --git a/requirements_torch_training.txt b/requirements_torch_training.txt
index 89e577db82ad5da5501ae1afee7946a490c001f4..c2b65e16a361157ef265ec9412025da516531334 100644
--- a/requirements_torch_training.txt
+++ b/requirements_torch_training.txt
@@ -1,4 +1,4 @@
-#git+https://gitlab.aicrowd.com/flatland/flatland.git@master
+git+https://gitlab.aicrowd.com/flatland/flatland.git@42-run-baselines-in-ci
 importlib-metadata>=0.17
 importlib_resources>=1.0.2
 torch>=1.1.0
\ No newline at end of file
diff --git a/torch_training/bla.py b/torch_training/bla.py
new file mode 100644
index 0000000000000000000000000000000000000000..5710167c1c3b3d248458a2a7ca6ce58832e6fe61
--- /dev/null
+++ b/torch_training/bla.py
@@ -0,0 +1 @@
+print("bla")
\ No newline at end of file
diff --git a/tox.ini b/tox.ini
index ffb9be07b29624769f52a82710962196cd9376fe..36b7c10a081b629375b202ad1e642e7366db2abe 100644
--- a/tox.ini
+++ b/tox.ini
@@ -20,10 +20,10 @@ passenv =
     HTTP_PROXY
     HTTPS_PROXY
 deps =
-;    -r{toxinidir}/requirements_torch_training.txt
+    -r{toxinidir}/requirements_torch_training.txt
 commands =
     python -m pip install -r requirements_torch_training.txt
-    python torch_training/multi_agent_training.py --n_trials=10
+    python torch_training/bla.py --n_trials=10
 
 [flake8]
 max-line-length = 120