diff --git a/libs/langgraph/langgraph/func/__init__.py b/libs/langgraph/langgraph/func/__init__.py
index f987081ca..2a54b7133 100644
--- a/libs/langgraph/langgraph/func/__init__.py
+++ b/libs/langgraph/langgraph/func/__init__.py
@@ -189,7 +189,7 @@ class EntrypointPregel(Pregel):
         self,
         config: Optional[RunnableConfig] = None,
         *,
-        xray: int | bool = False,
+        xray: Union[int, bool] = False,
     ) -> Graph:
         name, entrypoint = next(iter(self.nodes.items()))
         graph = Graph()