diff --git a/mhcflurry/train_allele_specific_models_command.py b/mhcflurry/train_allele_specific_models_command.py
index 8646a0208547ed6f241f84496c84b1a568e91ae5..bf8d653c868010c04506e26a02f2e50332cb542e 100644
--- a/mhcflurry/train_allele_specific_models_command.py
+++ b/mhcflurry/train_allele_specific_models_command.py
@@ -275,7 +275,7 @@ def run(argv=sys.argv[1:]):
             results = (
                 calibrate_percentile_ranks(
                     allele=allele,
-                    predictor=args.out_model_dir,
+                    predictor=predictor,
                     peptides=encoded_peptides)
                 for allele in alleles)
         else:
@@ -292,7 +292,7 @@ def run(argv=sys.argv[1:]):
             results = worker_pool.imap_unordered(
                 partial(
                     calibrate_percentile_ranks,
-                    predictor=predictor),
+                    predictor=args.out_models_dir),
                 alleles,
                 chunksize=1)