diff --git a/notebooks/unit1/requirements-unit1.txt b/notebooks/unit1/requirements-unit1.txt
index f95f064..cfb6770 100644
--- a/notebooks/unit1/requirements-unit1.txt
+++ b/notebooks/unit1/requirements-unit1.txt
@@ -1,4 +1,3 @@
-stable-baselines3[extra]
-box2d-py
+stable-baselines3==2.0.0a5
+gymnasium[box2d]
 huggingface_sb3
-pyglet==1.5.1