diff --git a/expected_workflows/flow_hybrid_expected.py b/expected_workflows/flow_hybrid_expected.py
index 5c92307..87d59d3 100644
--- a/expected_workflows/flow_hybrid_expected.py
+++ b/expected_workflows/flow_hybrid_expected.py
@@ -242,6 +242,7 @@ class test_repo_test_branch:
                 if isinstance(result, Exception):
                     logger.error(f"Task failed with exception: {result}")
                     workflow_output["status"] = "failed"
+            await temporalio.workflow.sleep(0)
             # Execution step 2
             tasks = [task_functions[node_id]() for node_id in ['m3aiq7ixuo6du35h8tr', 'm3aiqkrv4k1y6654ymr']]
             results_step = await asyncio.gather(*tasks, return_exceptions=True)
@@ -249,6 +250,7 @@ class test_repo_test_branch:
                 if isinstance(result, Exception):
                     logger.error(f"Task failed with exception: {result}")
                     workflow_output["status"] = "failed"
+            await temporalio.workflow.sleep(0)
 
             # Update workflow status to completed if not failed
             if workflow_output["status"] != "failed":
diff --git a/expected_workflows/flow_parallel_expected.py b/expected_workflows/flow_parallel_expected.py
index aba06e4..ef1d659 100644
--- a/expected_workflows/flow_parallel_expected.py
+++ b/expected_workflows/flow_parallel_expected.py
@@ -240,6 +240,7 @@ class test_repo_test_branch:
                 if isinstance(result, Exception):
                     logger.error(f"Task failed with exception: {result}")
                     workflow_output["status"] = "failed"
+            await temporalio.workflow.sleep(0)
 
             # Update workflow status to completed if not failed
             if workflow_output["status"] != "failed":
diff --git a/expected_workflows/flow_sequential_expected.py b/expected_workflows/flow_sequential_expected.py
index d0611ad..a6f338b 100644
--- a/expected_workflows/flow_sequential_expected.py
+++ b/expected_workflows/flow_sequential_expected.py
@@ -242,6 +242,7 @@ class test_repo_test_branch:
                 if isinstance(result, Exception):
                     logger.error(f"Task failed with exception: {result}")
                     workflow_output["status"] = "failed"
+            await temporalio.workflow.sleep(0)
             # Execution step 2
             tasks = [task_functions[node_id]() for node_id in ['m3aiq7ixuo6du35h8tr']]
             results_step = await asyncio.gather(*tasks, return_exceptions=True)
@@ -249,6 +250,7 @@ class test_repo_test_branch:
                 if isinstance(result, Exception):
                     logger.error(f"Task failed with exception: {result}")
                     workflow_output["status"] = "failed"
+            await temporalio.workflow.sleep(0)
             # Execution step 3
             tasks = [task_functions[node_id]() for node_id in ['m3aiqkrv4k1y6654ymr']]
             results_step = await asyncio.gather(*tasks, return_exceptions=True)
@@ -256,6 +258,7 @@ class test_repo_test_branch:
                 if isinstance(result, Exception):
                     logger.error(f"Task failed with exception: {result}")
                     workflow_output["status"] = "failed"
+            await temporalio.workflow.sleep(0)
 
             # Update workflow status to completed if not failed
             if workflow_output["status"] != "failed":