diff --git a/expected_workflows/flow_hybrid_expected.py b/expected_workflows/flow_hybrid_expected.py
index e73a22e..5c92307 100644
--- a/expected_workflows/flow_hybrid_expected.py
+++ b/expected_workflows/flow_hybrid_expected.py
@@ -16,6 +16,7 @@ logger = logging.getLogger(__name__)
 class test_repo_test_branch:
     @temporalio.workflow.run
     async def run(self, root_inputs: Dict[str, Any]) -> Dict[str, Any]:
+        await temporalio.workflow.sleep(0)
         workflow_info = temporalio.workflow.info()
         workflow_output: Dict[str, Any] = {
             "workflow_id": workflow_info.workflow_id,
diff --git a/expected_workflows/flow_parallel_expected.py b/expected_workflows/flow_parallel_expected.py
index e7fc71d..aba06e4 100644
--- a/expected_workflows/flow_parallel_expected.py
+++ b/expected_workflows/flow_parallel_expected.py
@@ -16,6 +16,7 @@ logger = logging.getLogger(__name__)
 class test_repo_test_branch:
     @temporalio.workflow.run
     async def run(self, root_inputs: Dict[str, Any]) -> Dict[str, Any]:
+        await temporalio.workflow.sleep(0)
         workflow_info = temporalio.workflow.info()
         workflow_output: Dict[str, Any] = {
             "workflow_id": workflow_info.workflow_id,
diff --git a/expected_workflows/flow_sequential_expected.py b/expected_workflows/flow_sequential_expected.py
index ef29ec6..d0611ad 100644
--- a/expected_workflows/flow_sequential_expected.py
+++ b/expected_workflows/flow_sequential_expected.py
@@ -16,6 +16,7 @@ logger = logging.getLogger(__name__)
 class test_repo_test_branch:
     @temporalio.workflow.run
     async def run(self, root_inputs: Dict[str, Any]) -> Dict[str, Any]:
+        await temporalio.workflow.sleep(0)
         workflow_info = temporalio.workflow.info()
         workflow_output: Dict[str, Any] = {
             "workflow_id": workflow_info.workflow_id,