diff --git a/setup.py b/setup.py
index 8736aae1..9188a2d6 100644
--- a/setup.py
+++ b/setup.py
@@ -43,7 +43,7 @@ setup(name='cement',
         'configparser': [],
         'daemon': [],
         'dummy': [],
-        'generate': [],
+        'generate': ['pyYaml'],
         'jinja2': ['jinja2'],
         'json': [],
         'logging': [],