diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 9d2b345cf42017a2f9c1043fd1dd1eedfd1aff4f..ea614dffc6e75effce8b60fd9224a9373595abfb 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -16,7 +16,7 @@ pages:
     - apt-get update -qy
     - apt-get install -y python3-dev python3-pip
     - pip install pdoc
-    - pip install .
+    - pip install ".[rl]"
     - pdoc --output-dir public overcooked_simulator !overcooked_simulator.reinforcement_learning --logo https://gitlab.ub.uni-bielefeld.de/uploads/-/system/project/avatar/6780/Cooking-Vector-Illustration-Icon-Graphics-4267218-1-580x435.jpg --docformat google
   artifacts:
     paths:
diff --git a/setup.py b/setup.py
index a4b8f900e5aa215bba1ef9b2d66078e5ce9096ab..20cbb15b660d46b5db2afd26b73045c486da0602 100644
--- a/setup.py
+++ b/setup.py
@@ -57,7 +57,7 @@ setup(
     url="https://gitlab.ub.uni-bielefeld.de/scs/cocosy/overcooked-simulator",
     version="0.1.0",
     zip_safe=False,
-    extras_requires={
+    extras_require={
         "rl": [
             "gymnasium>=0.28.1",
             "stable-baselines3[extra]>=2.2.1",