diff --git a/notebooks/unit1/requirements-unit1.txt b/notebooks/unit1/requirements-unit1.txt
index fd27dfc1..f74ac2a5 100644
--- a/notebooks/unit1/requirements-unit1.txt
+++ b/notebooks/unit1/requirements-unit1.txt
@@ -1,4 +1,5 @@
 stable-baselines3==2.0.0a5
 swig
-gymnasium[box2d]
+gymnasium
+box2d-py
 huggingface_sb3
diff --git a/notebooks/unit2/requirements-unit2.txt b/notebooks/unit2/requirements-unit2.txt
index 3209a69c..b30f2571 100644
--- a/notebooks/unit2/requirements-unit2.txt
+++ b/notebooks/unit2/requirements-unit2.txt
@@ -3,9 +3,9 @@ pygame
 numpy
 
 huggingface_hub
-pickle5
-pyyaml==6.0
+# pickle5 # `pickle5` installation fails on colab. according to https://github.com/huggingface/deep-rl-class/issues/616 using native `pickle` works fine
+pyyaml==6.0.2
 imageio
 imageio_ffmpeg
 pyglet==1.5.1
-tqdm
\ No newline at end of file
+tqdm