diff --git a/expected_workflows/flow_hybrid_expected.py b/expected_workflows/flow_hybrid_expected.py
index c45d802..d258365 100644
--- a/expected_workflows/flow_hybrid_expected.py
+++ b/expected_workflows/flow_hybrid_expected.py
@@ -6,7 +6,7 @@ import json
 import datetime
 import re
 import jmespath
-from temporalio import workflow
+from temporalio.workflow import sleep
 
 # Configure logging
 logging.basicConfig(level=logging.INFO,
@@ -17,7 +17,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 workflow.sleep(0)
+        await 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 1c9f407..50cf3ad 100644
--- a/expected_workflows/flow_parallel_expected.py
+++ b/expected_workflows/flow_parallel_expected.py
@@ -6,7 +6,7 @@ import json
 import datetime
 import re
 import jmespath
-from temporalio import workflow
+from temporalio.workflow import sleep
 
 # Configure logging
 logging.basicConfig(level=logging.INFO,
@@ -17,7 +17,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 workflow.sleep(0)
+        await 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 128fcba..7633797 100644
--- a/expected_workflows/flow_sequential_expected.py
+++ b/expected_workflows/flow_sequential_expected.py
@@ -6,7 +6,7 @@ import json
 import datetime
 import re
 import jmespath
-from temporalio import workflow
+from temporalio.workflow import sleep
 
 # Configure logging
 logging.basicConfig(level=logging.INFO,
@@ -17,7 +17,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 workflow.sleep(0)
+        await sleep(0)
         workflow_info = temporalio.workflow.info()
         workflow_output: Dict[str, Any] = {
             "workflow_id": workflow_info.workflow_id,