diff --git a/docker_scripts/parallelrunner.py b/docker_scripts/parallelrunner.py
index edc0991..ae1123a 100755
--- a/docker_scripts/parallelrunner.py
+++ b/docker_scripts/parallelrunner.py
@@ -326,7 +326,7 @@ def run_job(self, job_inputs):
 
         return job_outputs, status
 
-    def process_job_outputs(self, results, batch):
+    def process_job_outputs(self, results, batch, status):
         if self.template_id == "TEST_UUID":
             logger.info("Map in test mode, just returning input")
 
@@ -334,6 +334,7 @@ def process_job_outputs(self, results, batch):
 
         for task_i, task in enumerate(batch):
             output = task["output"]
+            task["status"] = status
             for probe_name, probe_outputs in results.items():
                 if probe_name not in output:
                     raise ValueError(f"Unknown probe in output: {probe_name}")
@@ -433,7 +434,7 @@ def map_func(batch, trial_number=1):
 
                 job_outputs, status = self.run_job(job_inputs)
 
-                batch = self.process_job_outputs(job_outputs, batch)
+                batch = self.process_job_outputs(job_outputs, batch, status)
 
                 logger.info(
                     "Worker has finished batch "