diff --git a/libs/langgraph/tests/test_pregel.py b/libs/langgraph/tests/test_pregel.py
index 196fe4223..89088bd1d 100644
--- a/libs/langgraph/tests/test_pregel.py
+++ b/libs/langgraph/tests/test_pregel.py
@@ -6996,11 +6996,14 @@ def test_branch_then(
     # test stream_mode=debug
     tool_two = tool_two_graph.compile(checkpointer=checkpointer)
     thread10 = {"configurable": {"thread_id": "10"}}
-    assert [
+    
+    res = [
         *tool_two.stream(
             {"my_key": "value", "market": "DE"}, thread10, stream_mode="debug"
         )
-    ] == [
+    ]
+    
+    assert res == [
         {
             "type": "checkpoint",
             "timestamp": AnyStr(),
@@ -7026,7 +7029,7 @@ def test_branch_then(
                 },
                 "parent_config": None,
                 "next": ["__start__"],
-                "tasks": [{"id": AnyStr(), "name": "__start__", "interrupts": ()}],
+                "tasks": [{"id": AnyStr(), "name": "__start__", "interrupts": (), "state": None}],
             },
         },
         {
@@ -7067,7 +7070,7 @@ def test_branch_then(
                     },
                 },
                 "next": ["prepare"],
-                "tasks": [{"id": AnyStr(), "name": "prepare", "interrupts": ()}],
+                "tasks": [{"id": AnyStr(), "name": "prepare", "interrupts": (), "state": None}],
             },
         },
         {
@@ -7131,7 +7134,7 @@ def test_branch_then(
                     },
                 },
                 "next": ["tool_two_slow"],
-                "tasks": [{"id": AnyStr(), "name": "tool_two_slow", "interrupts": ()}],
+                "tasks": [{"id": AnyStr(), "name": "tool_two_slow", "interrupts": (), "state": None}],
             },
         },
         {
@@ -7195,7 +7198,7 @@ def test_branch_then(
                     },
                 },
                 "next": ["finish"],
-                "tasks": [{"id": AnyStr(), "name": "finish", "interrupts": ()}],
+                "tasks": [{"id": AnyStr(), "name": "finish", "interrupts": (), "state": None}],
             },
         },
         {
diff --git a/libs/langgraph/tests/test_pregel_async.py b/libs/langgraph/tests/test_pregel_async.py
index 0290bf069..2c938b543 100644
--- a/libs/langgraph/tests/test_pregel_async.py
+++ b/libs/langgraph/tests/test_pregel_async.py
@@ -5655,7 +5655,14 @@ async def test_branch_then(checkpointer_name: str) -> None:
                     },
                     "parent_config": None,
                     "next": ["__start__"],
-                    "tasks": [{"id": AnyStr(), "name": "__start__", "interrupts": ()}],
+                    "tasks": [
+                        {
+                            "id": AnyStr(),
+                            "name": "__start__",
+                            "interrupts": (),
+                            "state": None,
+                        }
+                    ],
                 },
             },
             {
@@ -5696,7 +5703,14 @@ async def test_branch_then(checkpointer_name: str) -> None:
                         },
                     },
                     "next": ["prepare"],
-                    "tasks": [{"id": AnyStr(), "name": "prepare", "interrupts": ()}],
+                    "tasks": [
+                        {
+                            "id": AnyStr(),
+                            "name": "prepare",
+                            "interrupts": (),
+                            "state": None,
+                        }
+                    ],
                 },
             },
             {
@@ -5761,7 +5775,12 @@ async def test_branch_then(checkpointer_name: str) -> None:
                     },
                     "next": ["tool_two_slow"],
                     "tasks": [
-                        {"id": AnyStr(), "name": "tool_two_slow", "interrupts": ()}
+                        {
+                            "id": AnyStr(),
+                            "name": "tool_two_slow",
+                            "interrupts": (),
+                            "state": None,
+                        }
                     ],
                 },
             },
@@ -5826,7 +5845,14 @@ async def test_branch_then(checkpointer_name: str) -> None:
                         },
                     },
                     "next": ["finish"],
-                    "tasks": [{"id": AnyStr(), "name": "finish", "interrupts": ()}],
+                    "tasks": [
+                        {
+                            "id": AnyStr(),
+                            "name": "finish",
+                            "interrupts": (),
+                            "state": None,
+                        }
+                    ],
                 },
             },
             {
@@ -5937,7 +5963,14 @@ async def test_branch_then(checkpointer_name: str) -> None:
                     },
                     "parent_config": None,
                     "next": ["__start__"],
-                    "tasks": [{"id": AnyStr(), "name": "__start__", "interrupts": ()}],
+                    "tasks": [
+                        {
+                            "id": AnyStr(),
+                            "name": "__start__",
+                            "interrupts": (),
+                            "state": None,
+                        }
+                    ],
                 },
             },
             {
@@ -5978,7 +6011,14 @@ async def test_branch_then(checkpointer_name: str) -> None:
                         },
                     },
                     "next": ["prepare"],
-                    "tasks": [{"id": AnyStr(), "name": "prepare", "interrupts": ()}],
+                    "tasks": [
+                        {
+                            "id": AnyStr(),
+                            "name": "prepare",
+                            "interrupts": (),
+                            "state": None,
+                        }
+                    ],
                 },
             },
             {
@@ -6043,7 +6083,12 @@ async def test_branch_then(checkpointer_name: str) -> None:
                     },
                     "next": ["tool_two_slow"],
                     "tasks": [
-                        {"id": AnyStr(), "name": "tool_two_slow", "interrupts": ()}
+                        {
+                            "id": AnyStr(),
+                            "name": "tool_two_slow",
+                            "interrupts": (),
+                            "state": None,
+                        }
                     ],
                 },
             },