diff --git a/mhcflurry/train_allele_specific_models_command.py b/mhcflurry/train_allele_specific_models_command.py
index 1711a8a2097a52ed575ecf428de6eabe88c88e00..91dcedfc129523052c83c9287f09fcb8a3cd0cf9 100644
--- a/mhcflurry/train_allele_specific_models_command.py
+++ b/mhcflurry/train_allele_specific_models_command.py
@@ -309,7 +309,7 @@ def run(argv=sys.argv[1:]):
         # as it goes so no saving is required at the end.
         for _ in tqdm.trange(len(work_items)):
             item = work_items.pop(0)  # want to keep freeing up memory
-            (_, work_predictor) = train_model_entrypoint(item)
+            work_predictor = train_model_entrypoint(item)
             assert work_predictor is predictor
         assert not work_items