diff --git a/requirements.txt b/requirements.txt
index 33597dc..2ba1116 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -13,7 +13,7 @@ networkx >=2.1, <2.3
 numpy == 1.18.1
 pandas == 1.0.1
 papermill == 1.2.1
-PyYAML == 5.1
+PyYAML == 5.4
 scikit-learn == 0.22.2.post1
 scipy == 1.4.1
 seaborn == 0.9.0
diff --git a/setup.py b/setup.py
index d835fc3..6c1564d 100644
--- a/setup.py
+++ b/setup.py
@@ -65,7 +65,7 @@
     'numpy==1.18.1',
     'pandas==1.0.1',
     'papermill==1.2.1',
-    'PyYAML==5.1',
+    'PyYAML==5.4',
     'scikit-learn==0.22.2.post1',
     'scipy==1.4.1',
     'seaborn==0.9.0'