diff --git a/notebooks/unit2/requirements-unit2.txt b/notebooks/unit2/requirements-unit2.txt
index 3209a69c..f5614c3c 100644
--- a/notebooks/unit2/requirements-unit2.txt
+++ b/notebooks/unit2/requirements-unit2.txt
@@ -3,8 +3,7 @@ pygame
 numpy
 
 huggingface_hub
-pickle5
-pyyaml==6.0
+pyyaml>=6.0.1
 imageio
 imageio_ffmpeg
 pyglet==1.5.1
diff --git a/notebooks/unit2/unit2.ipynb b/notebooks/unit2/unit2.ipynb
index e9ae624c..db13d335 100644
--- a/notebooks/unit2/unit2.ipynb
+++ b/notebooks/unit2/unit2.ipynb
@@ -3,8 +3,7 @@
     {
       "cell_type": "markdown",
       "metadata": {
-        "id": "view-in-github",
-        "colab_type": "text"
+        "id": "view-in-github"
       },
       "source": [
         "<a href=\"https://colab.research.google.com/github/huggingface/deep-rl-class/blob/main/notebooks/unit2/unit2.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
@@ -289,7 +288,7 @@
         "import os\n",
         "import tqdm\n",
         "\n",
-        "import pickle5 as pickle\n",
+        "import pickle\n",
         "from tqdm.notebook import tqdm"
       ]
     },
@@ -1755,8 +1754,7 @@
         "B2_-8b8z5k54",
         "8R5ej1fS4P2V",
         "Pnpk2ePoem3r"
-      ],
-      "include_colab_link": true
+      ]
     },
     "gpuClass": "standard",
     "kernelspec": {