diff --git a/code/requirements.txt b/code/requirements.txt
index 66600fb..0efb677 100644
--- a/code/requirements.txt
+++ b/code/requirements.txt
@@ -1 +1 @@
-PyYaml==5.1
+PyYaml==5.4