diff --git a/MNIST/requirements.txt b/MNIST/requirements.txt
index 6b5eaa3..b0b155d 100644
--- a/MNIST/requirements.txt
+++ b/MNIST/requirements.txt
@@ -17,7 +17,7 @@ MarkupSafe==1.1.1
 numpy==1.16.4
 Pillow==6.0.0
 protobuf==3.8.0
-PyYAML==5.1.1
+PyYAML==5.4
 scipy==1.3.0
 six==1.12.0
 tensorboard==1.9.0