diff --git a/mhcflurry/downloads.yml b/mhcflurry/downloads.yml
index 4c01ebaab0cbbc51cd231d50532abbf84ba13f4d..0b07827b79806fe677d8e2d30249fa5402c9c492 100644
--- a/mhcflurry/downloads.yml
+++ b/mhcflurry/downloads.yml
@@ -21,7 +21,11 @@ releases:
         compatibility-version: 2
         downloads:
             - name: models_class1
-              url: http://github.com/openvax/mhcflurry/releases/download/pre-1.1/models_class1.20180205.tar.bz2
+              url: https://github.com/openvax/mhcflurry/releases/download/pre-1.2/models_class1.20180222.tar.bz2
+              default: true
+
+            - name: models_class1_selected_no_mass_spec
+              url: https://github.com/openvax/mhcflurry/releases/download/pre-1.2/models_class1_selected_no_mass_spec.20180222.tar.bz2
               default: true
 
             - name: models_class1_unselected
diff --git a/test/test_train_and_related_commands.py b/test/test_train_and_related_commands.py
index 0e8e3d7645c1e781963c5357285c80fff747ac48..ef5664af5c23cb4fc44abf54a5f9c6f50720531c 100644
--- a/test/test_train_and_related_commands.py
+++ b/test/test_train_and_related_commands.py
@@ -134,7 +134,7 @@ def run_and_check_with_model_selection(n_jobs=1):
         "--out-models-dir", models_dir2,
         "--models-dir", models_dir1,
         "--num-jobs", str(n_jobs),
-        "--max-models", "1"
+        "--mse-max-models", "1"
     ]
     print("Running with args: %s" % args)
     subprocess.check_call(args)