diff --git a/rl_unplugged/dm_control_suite_d4pg.ipynb b/rl_unplugged/dm_control_suite_d4pg.ipynb
index 1c8ed6c..f401fda 100644
--- a/rl_unplugged/dm_control_suite_d4pg.ipynb
+++ b/rl_unplugged/dm_control_suite_d4pg.ipynb
@@ -119,9 +119,8 @@
         "# Configure dm_control to use the OSMesa rendering backend\n",
         "%env MUJOCO_GL=osmesa\n",
         "\n",
-        "# Install dm_control, including extra dependencies needed for the locomotion\n",
-        "# mazes.\n",
-        "!pip install dm_control[locomotion_mazes]"
+        "# Install dm_control\n",
+        "!pip install dm_control"
       ]
     },
     {
diff --git a/rl_unplugged/rwrl_d4pg.ipynb b/rl_unplugged/rwrl_d4pg.ipynb
index 3687ca8..672b9ee 100644
--- a/rl_unplugged/rwrl_d4pg.ipynb
+++ b/rl_unplugged/rwrl_d4pg.ipynb
@@ -116,9 +116,8 @@
         "# Configure dm_control to use the OSMesa rendering backend\n",
         "%env MUJOCO_GL=osmesa\n",
         "\n",
-        "# Install dm_control, including extra dependencies needed for the locomotion\n",
-        "# mazes.\n",
-        "!pip install dm_control[locomotion_mazes]"
+        "# Install dm_control\n",
+        "!pip install dm_control"
       ]
     },
     {