diff --git a/utilities/aws_lambda/task_handler.py b/utilities/aws_lambda/task_handler.py
index 0ed680c92..ada8562ec 100644
--- a/utilities/aws_lambda/task_handler.py
+++ b/utilities/aws_lambda/task_handler.py
@@ -66,9 +66,15 @@ def task_handler(
                     )
 
                     try:
+                        if pass_task_orchestrator:
+                            work = lambda body=body, t=task: func(
+                                body, context, orchestrator, t.id
+                            )
+                        else:
+                            work = lambda body=body: func(body, context)
                         orchestrator.run_subtask(
                             subtask.id,
-                            work=lambda body=body: func(body, context),
+                            work=work,
                             cloud_logs_url=cloud_logs_url,
                         )
                     except Exception: