diff --git a/benchmark/MHD/generate-experiment.py b/benchmark/MHD/generate-experiment.py
index f7bd3b4..e07ad52 100644
--- a/benchmark/MHD/generate-experiment.py
+++ b/benchmark/MHD/generate-experiment.py
@@ -175,7 +175,7 @@ def generate_strong_scaling_experiments(
 def configure_cparams(cparams_file: pathlib.Path, experiment: Experiment):
     cparams = [
         f"integer, parameter :: ncpus={experiment.gpus}\n",
-        f"integer, parameter :: nprocx={experiment.nproc[0]},nprocy={experiment.nproc[1]},nprocz={experiment.nproc[1]}\n",
+        f"integer, parameter :: nprocx={experiment.nproc[0]},nprocy={experiment.nproc[1]},nprocz={experiment.nproc[2]}\n",
         f"integer, parameter :: nxgrid={experiment.ngrid[0]},nygrid={experiment.ngrid[1]},nzgrid={experiment.ngrid[2]}\n",
     ]
 
@@ -262,6 +262,13 @@ def generate(
             target = os.path.join(template_dir, "src", f.name)
             shutil.copy2(f, target)
 
+        f = os.path.join(sample_path, "src", "astaroth", "DSL", "local", "equations.h")
+        target = os.path.join(
+            template_dir, "src", "astaroth", "DSL", "local", "equations.h"
+        )
+        os.makedirs(os.path.dirname(target))
+        shutil.copy2(f, target)
+
         # Create the experiments
         for experiment in experiments:
             print(f"Creating experiment '{experiment.name}'")