diff --git a/src/requirements.txt b/src/requirements.txt
index 7a0ef67..389f68b 100644
--- a/src/requirements.txt
+++ b/src/requirements.txt
@@ -7,4 +7,4 @@ torchvision==0.2.2.post3
 opencv_contrib_python==4.1.0.25
 matplotlib==2.2.3
 Pillow==6.1.0
-PyYAML==5.1.1
+PyYAML==5.4