diff --git a/downloads-generation/data_mass_spec_benchmark/run_mhcflurry.py b/downloads-generation/data_mass_spec_benchmark/run_mhcflurry.py
index 804cd531d0ecdcd97b548737d89641fb812899e9..5e2f6d3d2564e1122541de90f115810c5b025e4a 100644
--- a/downloads-generation/data_mass_spec_benchmark/run_mhcflurry.py
+++ b/downloads-generation/data_mass_spec_benchmark/run_mhcflurry.py
@@ -6,7 +6,6 @@ import signal
 import sys
 import time
 import traceback
-import collections
 import math
 from functools import partial
 
@@ -231,6 +230,10 @@ def run(argv=sys.argv[1:]):
 
 
 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
+
     if constant_data is None:
         constant_data = GLOBAL_DATA