diff --git a/requirements.txt b/requirements.txt
index 1b55681..fbe3d1d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,4 +2,4 @@ tensorflow_transform==0.13.0
 pandas==0.24.2
 tensorflow==2.4.0
 apache_beam==2.13.0
-PyYAML==5.1.1
+PyYAML==5.4