diff --git a/notebooks/unit3/unit3.ipynb b/notebooks/unit3/unit3.ipynb
index 5c21dca..8a4711c 100644
--- a/notebooks/unit3/unit3.ipynb
+++ b/notebooks/unit3/unit3.ipynb
@@ -289,7 +289,9 @@
       },
       "outputs": [],
       "source": [
-        "!pip install -r requirements.txt"
+        "!pip install -r requirements.txt",
+        "# Since colab uses Python 3.9 we need to add this installation\n",
+        "!pip install gym[atari,accept-rom-license]==0.21.0"
       ]
     },
     {