diff --git a/notebooks/unit2/requirements-unit2.txt b/notebooks/unit2/requirements-unit2.txt
index 3209a69c..0389f3fc 100644
--- a/notebooks/unit2/requirements-unit2.txt
+++ b/notebooks/unit2/requirements-unit2.txt
@@ -4,8 +4,8 @@ numpy
 
 huggingface_hub
 pickle5
-pyyaml==6.0
+pyyaml>=6.0.1
 imageio
 imageio_ffmpeg
 pyglet==1.5.1
-tqdm
\ No newline at end of file
+tqdm