diff --git a/notebooks/unit1/requirements-unit1.txt b/notebooks/unit1/requirements-unit1.txt
index fd27dfc1..70961ac6 100644
--- a/notebooks/unit1/requirements-unit1.txt
+++ b/notebooks/unit1/requirements-unit1.txt
@@ -1,4 +1,5 @@
-stable-baselines3==2.0.0a5
+stable-baselines3
 swig
-gymnasium[box2d]
+gymnasium
+box2d-py
 huggingface_sb3