diff --git a/keras-assignment/requirements.txt b/keras-assignment/requirements.txt
index 7c0c218..82250c6 100644
--- a/keras-assignment/requirements.txt
+++ b/keras-assignment/requirements.txt
@@ -10,7 +10,7 @@ Keras-Preprocessing==1.0.9
 Markdown==3.1.1
 numpy==1.16.3
 protobuf==3.7.1
-PyYAML==5.1
+PyYAML==5.4
 scipy==1.3.0
 six==1.12.0
 tensorboard==1.12.2