diff --git a/libs/langgraph/langgraph/pregel/__init__.py b/libs/langgraph/langgraph/pregel/__init__.py
index 1cf5ec982..671b6f843 100644
--- a/libs/langgraph/langgraph/pregel/__init__.py
+++ b/libs/langgraph/langgraph/pregel/__init__.py
@@ -1245,7 +1245,8 @@ class Pregel(Runnable[Union[dict[str, Any], Any], Union[dict[str, Any], Any]]):
                     def get_waiter() -> asyncio.Task[None]:
                         nonlocal waiter
                         if waiter is None or waiter.done():
-                            return (waiter := loop.submit(stream.wait))
+                            waiter = loop.submit(stream.wait)
+                            return waiter
                         else:
                             return waiter
                 else: