diff --git a/assignment2/requirements.txt b/assignment2/requirements.txt
index 046234c..2e9d002 100644
--- a/assignment2/requirements.txt
+++ b/assignment2/requirements.txt
@@ -51,7 +51,7 @@ Pygments==2.3.1
 pyparsing==2.4.0
 pyrsistent==0.14.11
 python-dateutil==2.8.0
-PyYAML==5.1
+PyYAML==5.4
 pyzmq==18.0.1
 qtconsole==4.4.3
 scipy==1.2.1