diff --git a/tests/worker/test_workflow.py b/tests/worker/test_workflow.py index cfc0190314..8594988ae9 100644 --- a/tests/worker/test_workflow.py +++ b/tests/worker/test_workflow.py @@ -1340,7 +1340,7 @@ async def run(self) -> NoReturn: except asyncio.CancelledError: self._reason_inner = workflow.cancellation_reason() try: - await asyncio.sleep(1000) + await workflow.wait_condition(lambda: False) except asyncio.CancelledError: self._reason_outer = workflow.cancellation_reason() raise