diff --git a/packages/runtime-sdk/src/asgi.py b/packages/runtime-sdk/src/asgi.py
index a056fcb..705615f 100644
--- a/packages/runtime-sdk/src/asgi.py
+++ b/packages/runtime-sdk/src/asgi.py
@@ -7,7 +7,7 @@
 
 import js
 
-from workers import Context, Request, wait_until
+from workers import Context, Request
 
 ASGI = {"spec_version": "2.0", "version": "3.0"}
 logger = logging.getLogger("asgi")
@@ -221,6 +221,9 @@ async def run_app():
 
     # Create task to run the application in the background
     app_task = create_proxy(create_task(run_app()))
+
+    from workers import wait_until
+
     wait_until(app_task)
 
     try: