diff --git a/downloads-generation/models_class1_unselected/GENERATE.sh b/downloads-generation/models_class1_unselected/GENERATE.sh
index f9f3545cf868663076704c739a848239703c5389..a3cdb5c0fa5e36e88407e1caacc5b22bb2c0f042 100755
--- a/downloads-generation/models_class1_unselected/GENERATE.sh
+++ b/downloads-generation/models_class1_unselected/GENERATE.sh
@@ -38,7 +38,6 @@ PROCESSORS=$(getconf _NPROCESSORS_ONLN)
 echo "Detected processors: $PROCESSORS"
 
 time mhcflurry-class1-train-allele-specific-models \
-    --allele HLA-A*02:01 HLA-A*01:01 HLA-A*03:01 HLA-A*11:01 HLA-A*24:02 HLA-B*07:02 HLA-B*15:01 \
     --data "$(mhcflurry-downloads path data_curated)/curated_training_data.no_mass_spec.csv.bz2" \
     --hyperparameters hyperparameters.yaml \
     --out-models-dir models \
diff --git a/mhcflurry/class1_neural_network.py b/mhcflurry/class1_neural_network.py
index 4bc8331b165bbee85ddc4c3272659fdf25bab1c3..9b2841227b95855dfdec328fd26a9376a58df19e 100644
--- a/mhcflurry/class1_neural_network.py
+++ b/mhcflurry/class1_neural_network.py
@@ -103,7 +103,7 @@ class Class1NeuralNetwork(object):
     """
 
     miscelaneous_hyperparameter_defaults = HyperparameterDefaults(
-        train_data={'subset': 'all', 'num_poitns': None},
+        train_data={'subset': 'all'},
     )
     """
     Miscelaneous hyperaparameters. These parameters are not used by this class