diff --git a/run_modal.py b/run_modal.py
index 4675c1cb..a1243129 100644
--- a/run_modal.py
+++ b/run_modal.py
@@ -85,7 +85,6 @@
     torch.autograd.set_detect_anomaly(True)
 
 import argparse
-from toolkit.job import get_job
 
 def print_end_message(jobs_completed, jobs_failed):
     failure_string = f"{jobs_failed} failure{'' if jobs_failed == 1 else 's'}" if jobs_failed > 0 else ""
@@ -109,6 +108,7 @@ def print_end_message(jobs_completed, jobs_failed):
     timeout=7200  # 2 hours, increase or decrease if needed
 )
 def main(config_file_list_str: str, recover: bool = False, name: str = None):
+    from toolkit.job import get_job
     # convert the config file list from a string to a list
     config_file_list = config_file_list_str.split(",")