diff --git a/.travis.yml b/.travis.yml
index 45dd1f9f5b72fc5c5821c2d903ac9555260793de..860401d0098145a3da4fdc877e9751b72cd1f32f 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -39,13 +39,14 @@ script:
   # download training data
   - script/download-kim-2013-dataset.sh
   - script/download-iedb.sh
-  - script/create-iedb-class1-dataset.py
+  # only install data for A*01:01 and A*02:01 for testing
+  - script/create-iedb-class1-dataset.py --alleles HLA-A*01:01 HLA-A*02:01
   - script/create-combined-class1-dataset.py
   # 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
+      --alleles HLA-A0101 HLA-A0201
       --embedding-size 4
       --hidden-layer-size 4
       --training-epochs 5