diff --git a/.travis.yml b/.travis.yml
index a1d822243017d5211e8dbfdc2e2257a81ae47a84..f5788e9fddd01c8d160be76100d48e914676f3a9 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -1,5 +1,5 @@
 language: python
-install: pip install pyyaml
+install: pip install --requirement tools/generation/requirements.txt
 script:
   - echo The test generation tool should be working.
   - ./tools/generation/test/run.py