diff --git a/downloads-generation/data_mass_spec_benchmark/run_predictors.py b/downloads-generation/data_mass_spec_benchmark/run_predictors.py
index be034ba9b37ccb7b7a5b1be99785564104c08c2d..af6a7868ed0e70ac54fe51cd93f4be6a60c90ca4 100644
--- a/downloads-generation/data_mass_spec_benchmark/run_predictors.py
+++ b/downloads-generation/data_mass_spec_benchmark/run_predictors.py
@@ -234,7 +234,7 @@ def run(argv=sys.argv[1:]):
             peptide_chunks = numpy.array_split(peptides, num_chunks)
             for chunk_peptides in peptide_chunks:
                 work_items.append({
-                    'alleles': alleles[list(allele_indices_to_peptides)],
+                    'alleles': alleles[list(indices)],
                     'peptides': chunk_peptides,
                 })
     else: