diff --git a/client/verta/setup.py b/client/verta/setup.py
index 92ce76ae62..a193693bd5 100644
--- a/client/verta/setup.py
+++ b/client/verta/setup.py
@@ -31,7 +31,7 @@
         "googleapis-common-protos>=1.5",
         "pathlib2>=2.2",
         "protobuf>=3.8",
-        "pyyaml>=5.1",
+        "pyyaml>=5.1, <5.4",
         "requests>=2.21, <3.0",
     ],
     entry_points={
diff --git a/client/verta/verta/client.py b/client/verta/verta/client.py
index c8fcd80c61..21b1fc9cff 100644
--- a/client/verta/verta/client.py
+++ b/client/verta/verta/client.py
@@ -19,7 +19,6 @@
 import zipfile
 
 import requests
-import yaml
 from verta._tracking.organization import Organization
 from ._internal_utils._utils import check_unnecessary_params_warning