diff --git a/17_reinforcement_learning.ipynb b/17_reinforcement_learning.ipynb
index 334d1a3..12f185e 100644
--- a/17_reinforcement_learning.ipynb
+++ b/17_reinforcement_learning.ipynb
@@ -58,8 +58,9 @@
     "\n",
     "if IS_COLAB or IS_KAGGLE:\n",
     "    !apt update && apt install -y libpq-dev libsdl2-dev swig xorg-dev xvfb\n",
-    "    %pip install -q -U tf-agents pyvirtualdisplay gym[box2d,atari_py] AutoROM\n",
-    "    AUTOROM_PREFIX=\"/bin\n",
+    "    %pip install -U tf-agents pyvirtualdisplay\n",
+    "    %pip install -U gym>=0.21.0\n",
+    "    %pip install -U gym[box2d,atari,accept-rom-license]\n",
     "\n",
     "# Common imports\n",
     "import numpy as np\n",