diff --git a/.travis.yml b/.travis.yml
index a1d8222430..f5788e9fdd 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