diff --git a/setup.py b/setup.py
index 2cc1d9d8..3624e381 100644
--- a/setup.py
+++ b/setup.py
@@ -11,6 +11,7 @@
     'pexpect',
     'packaging',
     'urllib3',
+    'pyyaml',
 ]
 
 # docker API used to be in a package called `docker-py` before the 2.0 release