diff --git a/setup.py b/setup.py
index <HASH>..<HASH> 100644
--- a/setup.py
+++ b/setup.py
@@ -30,7 +30,11 @@ INSTALL_REQUIRES = (
     'pywinrm',
 )
 
-TEST_REQUIRES = (
+ANSIBLE_REQUIRES = (  # extras for parsing Ansible inventory
+    'pyyaml',
+)
+
+TEST_REQUIRES = ANSIBLE_REQUIRES + (
     # Unit testing
     'pytest==4.6.6',
     'pytest-cov==2.8.1',
@@ -65,10 +69,6 @@ DEV_REQUIRES = TEST_REQUIRES + DOCS_REQUIRES + (
     'ipdbplugin==1.4.5',
 )
 
-ANSIBLE_REQUIRES = (
-    'pyyaml',
-)
-
 
 # Extract version info without importing entire pyinfra package
 version_data = {}
Include Ansible extras in test requirements.