diff --git a/downloads-generation/data_mass_spec_benchmark/run_mhcflurry.py b/downloads-generation/data_mass_spec_benchmark/run_mhcflurry.py
index 5e2f6d3d2564e1122541de90f115810c5b025e4a..5f0235cb17b8a84e48322df0305196cc9d9102bc 100644
--- a/downloads-generation/data_mass_spec_benchmark/run_mhcflurry.py
+++ b/downloads-generation/data_mass_spec_benchmark/run_mhcflurry.py
@@ -233,6 +233,7 @@ def do_predictions(chunk_index, peptides, alleles, constant_data=None):
     # This may run on the cluster in a way that misses all top level imports,
     # so we have to re-import everything here.
     import time
+    from mhcflurry.encodable_sequences import EncodableSequences
 
     if constant_data is None:
         constant_data = GLOBAL_DATA