diff --git a/notebooks/unit3/unit3.ipynb b/notebooks/unit3/unit3.ipynb
index 78f1ac7..2252762 100644
--- a/notebooks/unit3/unit3.ipynb
+++ b/notebooks/unit3/unit3.ipynb
@@ -278,7 +278,8 @@
       },
       "outputs": [],
       "source": [
-        "%cd /content/rl-baselines3-zoo/"
+        "%cd /content/rl-baselines3-zoo/ \n",
+        "!git checkout v1.8.0"
       ]
     },
     {