diff --git a/MANIFEST.in b/MANIFEST.in
index c3286295576ccdb7033587afbc434f097390a42a..15ae22aa6240e9fc05354c6be3fe120383a8f91c 100644
--- a/MANIFEST.in
+++ b/MANIFEST.in
@@ -1,8 +1,8 @@
-include AUTHORS.rst
+include AUTHORS.md
 include CONTRIBUTING.rst
-include HISTORY.rst
+include changelog.md
 include LICENSE
-include README.rst
+include README.md
 include requirements_torch_training.txt
 include requirements_RLLib_training.txt
 
@@ -12,4 +12,4 @@ recursive-include tests *
 recursive-exclude * __pycache__
 recursive-exclude * *.py[co]
 
-recursive-include docs *.rst *.md conf.py Makefile make.bat *.jpg *.png *.gif
+recursive-include docs *.rst *.md conf.py *.jpg *.png *.gif
diff --git a/requirements_torch_training.txt b/requirements_torch_training.txt
index c2b65e16a361157ef265ec9412025da516531334..0fc821c65cc2c79ca68f4fac7895da4bbc35471d 100644
--- a/requirements_torch_training.txt
+++ b/requirements_torch_training.txt
@@ -1,4 +1,4 @@
-git+https://gitlab.aicrowd.com/flatland/flatland.git@42-run-baselines-in-ci
+git+https://gitlab.aicrowd.com/flatland/flatland.git
 importlib-metadata>=0.17
 importlib_resources>=1.0.2
 torch>=1.1.0
\ No newline at end of file