diff --git a/benchmarks/insert_muon/Snakefile b/benchmarks/insert_muon/Snakefile
index 622733c..a9a3216 100644
--- a/benchmarks/insert_muon/Snakefile
+++ b/benchmarks/insert_muon/Snakefile
@@ -14,7 +14,8 @@ root -l -b -q '{input.script}({params.NEVENTS_GEN},"{output.GEN_FILE}", "mu-", {
 
 rule insert_muon_simulate:
 	input:
-		GEN_FILE="sim_output/insert_muon/mu-_{P}GeV.hepmc"
+                GEN_FILE="sim_output/insert_muon/mu-_{P}GeV.hepmc",
+                warmup="warmup/{DETECTOR_CONFIG}.edm4hep.root",
 	params:
 		PHYSICS_LIST="FTFP_BERT"
 	output:
diff --git a/benchmarks/insert_neutron/Snakefile b/benchmarks/insert_neutron/Snakefile
index d14a7be..136c56e 100644
--- a/benchmarks/insert_neutron/Snakefile
+++ b/benchmarks/insert_neutron/Snakefile
@@ -15,7 +15,8 @@ root -l -b -q '{input.script}({params.NEVENTS_GEN},"{output.GEN_FILE}", "neutron
 
 rule insert_neutron_simulate:
 	input:
-		GEN_FILE="sim_output/insert_neutron/neutron_{P}GeV.hepmc"
+                GEN_FILE="sim_output/insert_neutron/neutron_{P}GeV.hepmc",
+                warmup="warmup/{DETECTOR_CONFIG}.edm4hep.root",
 	params:
 		PHYSICS_LIST="FTFP_BERT"
 	output:
diff --git a/benchmarks/zdc_lambda/Snakefile b/benchmarks/zdc_lambda/Snakefile
index f7f934d..9ffd170 100644
--- a/benchmarks/zdc_lambda/Snakefile
+++ b/benchmarks/zdc_lambda/Snakefile
@@ -12,7 +12,8 @@ root -l -b -q '{input.script}({params.NEVENTS_GEN},0,"{output.GEN_FILE}",{wildca
 
 rule zdc_lambda_simulate:
 	input:
-		GEN_FILE="sim_output/zdc_lambda/lambda_decay_{P}GeV.hepmc"
+                GEN_FILE="sim_output/zdc_lambda/lambda_decay_{P}GeV.hepmc",
+                warmup="warmup/{DETECTOR_CONFIG}.edm4hep.root",
 	params:
 		PHYSICS_LIST="FTFP_BERT"
 	output:
diff --git a/benchmarks/zdc_photon/Snakefile b/benchmarks/zdc_photon/Snakefile
index 0c551ab..c8302c5 100644
--- a/benchmarks/zdc_photon/Snakefile
+++ b/benchmarks/zdc_photon/Snakefile
@@ -15,7 +15,8 @@ root -l -b -q '{input.script}('$NEVENTS_GEN',"{output.GEN_FILE}", "gamma", {para
 
 rule zdc_photon_simulate:
 	input:
-		GEN_FILE="sim_output/zdc_photon/zdc_photon_{P}GeV.hepmc"
+                GEN_FILE="sim_output/zdc_photon/zdc_photon_{P}GeV.hepmc",
+                warmup="warmup/{DETECTOR_CONFIG}.edm4hep.root",
 	params:
 		PHYSICS_LIST="FTFP_BERT"
 	output:
diff --git a/benchmarks/zdc_pi0/Snakefile b/benchmarks/zdc_pi0/Snakefile
index 8bc5c40..f751442 100644
--- a/benchmarks/zdc_pi0/Snakefile
+++ b/benchmarks/zdc_pi0/Snakefile
@@ -13,7 +13,8 @@ root -l -b -q '{input.script}({params.NEVENTS_GEN},0,"{output.GEN_FILE}",{wildca
 
 rule zdc_pi0_simulate:
 	input:
-		GEN_FILE="sim_output/zdc_pi0/zdc_pi0_{P}GeV.hepmc"
+                GEN_FILE="sim_output/zdc_pi0/zdc_pi0_{P}GeV.hepmc",
+                warmup="warmup/{DETECTOR_CONFIG}.edm4hep.root",
 	params:
 		PHYSICS_LIST="FTFP_BERT"
 	output:
diff --git a/benchmarks/zdc_sigma/Snakefile b/benchmarks/zdc_sigma/Snakefile
index 6f6e4ab..80e8b25 100644
--- a/benchmarks/zdc_sigma/Snakefile
+++ b/benchmarks/zdc_sigma/Snakefile
@@ -12,7 +12,8 @@ root -l -b -q '{input.script}({params.NEVENTS_GEN},0,"{output.GEN_FILE}",{wildca
 
 rule zdc_sigma_simulate:
 	input:
-		GEN_FILE="sim_output/zdc_sigma/sigma_decay_{P}GeV.hepmc"
+                GEN_FILE="sim_output/zdc_sigma/sigma_decay_{P}GeV.hepmc",
+                warmup="warmup/{DETECTOR_CONFIG}.edm4hep.root",
 	params:
 		PHYSICS_LIST="FTFP_BERT"
 	output: