diff --git a/setup.py b/setup.py
index a45e50e..7ccbd6f 100644
--- a/setup.py
+++ b/setup.py
@@ -21,7 +21,7 @@
     "license": "Apache-2.0",
     "author_email": "jefvantongerloo@gmail.com",
     "include_package_data": True,
-    "install_requires": ["scrapli", "textfsm"],
+    "install_requires": ["scrapli", "textfsm", "pyyaml"],
     "packages": ["textfsm_aos"],
     "scripts": "",
     "zip_safe": True,