diff --git a/docker_scripts/parallelrunner.py b/docker_scripts/parallelrunner.py
index cdb5827..aabae9f 100755
--- a/docker_scripts/parallelrunner.py
+++ b/docker_scripts/parallelrunner.py
@@ -486,7 +486,7 @@ def map_func(batch_with_uuid, trial_number=1):
                         f"trial {trial_number}, retrying "
                     )
                     output_batch = map_func(
-                        batch, trial_number=trial_number + 1
+                        batch_with_uuid, trial_number=trial_number + 1
                     )
 
             return output_batch