diff --git a/.travis.yml b/.travis.yml
index 4a71970a21e56d5715e492f5b3f74d59940adc46..a6b3b506bfd9fff20d374d65df291a9085fb20ef 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -36,6 +36,14 @@ install:
   - pip install .
   - pip install coveralls
 script:
+  # only installing A0101 and A0201 for testing purposes and with very limited
+  # training
+  - >
+      mhcflurry-train-class1-allele-specific-models.py
+      --alleles HLA-A0101 HLA-A0101
+      --embedding-size 4
+      --hidden-layer-size 4
+      --training-epochs 5
   # run tests
   - nosetests test --with-coverage --cover-package=mhcflurry  && ./lint.sh
 after_success: