diff --git a/mdbenchmark/submit.py b/mdbenchmark/submit.py
index adf9ae52..bfaca3b6 100644
--- a/mdbenchmark/submit.py
+++ b/mdbenchmark/submit.py
@@ -125,7 +125,7 @@ def submit(directory, force_restart, yes):
     for sim in bundles_to_start:
         # Remove files generated by previous mdbenchmark run
         if force_restart:
-            engine = detect_md_engine(sim["module"])
+            engine = detect_md_engine(sim.categories["module"])
             cleanup_before_restart(engine=engine, sim=sim)
         sim.categories["started"] = True
         os.chdir(sim.abspath)