diff --git a/test/test_train_allele_specific_models_command.py b/test/test_train_allele_specific_models_command.py
index a1e597ca0e9b37069358e1e61e79ca2433a4c9f5..ab79e680fa9c0df773e3822378117ae19933765f 100644
--- a/test/test_train_allele_specific_models_command.py
+++ b/test/test_train_allele_specific_models_command.py
@@ -12,7 +12,7 @@ from mhcflurry.downloads import get_path
 HYPERPARAMETERS = [
     {
         "n_models": 2,
-        "max_epochs": 20,
+        "max_epochs": 100,
         "patience": 10,
         "early_stopping": True,
         "validation_split": 0.2,