diff --git a/code/requirements.txt b/code/requirements.txt
index 36fba38..99e1d4a 100644
--- a/code/requirements.txt
+++ b/code/requirements.txt
@@ -50,7 +50,7 @@ pyparsing==2.4.0
 python-dateutil==2.8.0
 pytz==2019.1
 PyWavelets==1.0.3
-PyYAML==5.1
+PyYAML==5.4
 quiver-engine==0.1.4.1.4
 scikit-image==0.14.5
 scikit-learn==0.20.4