diff --git a/notebooks/unit2/requirements-unit2.txt b/notebooks/unit2/requirements-unit2.txt
index 995b24b..3209a69 100644
--- a/notebooks/unit2/requirements-unit2.txt
+++ b/notebooks/unit2/requirements-unit2.txt
@@ -1,4 +1,4 @@
-gym==0.24
+gymnasium
 pygame
 numpy
 
@@ -8,4 +8,4 @@ pyyaml==6.0
 imageio
 imageio_ffmpeg
 pyglet==1.5.1
-tqdm
+tqdm
\ No newline at end of file