diff --git a/libs/scheduler-kafka/tests/test_subgraph.py b/libs/scheduler-kafka/tests/test_subgraph.py
index 934e3a614..4febf76f0 100644
--- a/libs/scheduler-kafka/tests/test_subgraph.py
+++ b/libs/scheduler-kafka/tests/test_subgraph.py
@@ -202,7 +202,7 @@ async def test_subgraph_w_interrupt(
                                 "subgraph_counter": None,
                                 "call_counter": None,
                                 "interrupt_counter": None,
-                                "null_resume": None,
+                                "get_null_resume": None,
                                 "resume": [],
                             },
                             "checkpoint_id": None,
@@ -275,7 +275,7 @@ async def test_subgraph_w_interrupt(
                                 "subgraph_counter": None,
                                 "call_counter": None,
                                 "interrupt_counter": None,
-                                "null_resume": None,
+                                "get_null_resume": None,
                                 "resume": [],
                             },
                             "checkpoint_id": c.config["configurable"]["checkpoint_id"],
@@ -378,7 +378,7 @@ async def test_subgraph_w_interrupt(
                                 "subgraph_counter": None,
                                 "call_counter": None,
                                 "interrupt_counter": None,
-                                "null_resume": None,
+                                "get_null_resume": None,
                                 "resume": [],
                             },
                             "checkpoint_id": c.config["configurable"]["checkpoint_id"],
@@ -491,7 +491,7 @@ async def test_subgraph_w_interrupt(
                                 "subgraph_counter": None,
                                 "call_counter": None,
                                 "interrupt_counter": None,
-                                "null_resume": None,
+                                "get_null_resume": None,
                                 "resume": [],
                             },
                             "checkpoint_id": None,
@@ -559,7 +559,7 @@ async def test_subgraph_w_interrupt(
                                 "subgraph_counter": None,
                                 "call_counter": None,
                                 "interrupt_counter": None,
-                                "null_resume": None,
+                                "get_null_resume": None,
                                 "resume": [],
                             },
                             "checkpoint_id": c.config["configurable"]["checkpoint_id"],
@@ -683,7 +683,7 @@ async def test_subgraph_w_interrupt(
                                 "subgraph_counter": None,
                                 "call_counter": None,
                                 "interrupt_counter": None,
-                                "null_resume": None,
+                                "get_null_resume": None,
                                 "resume": [],
                             },
                             "checkpoint_id": c.config["configurable"]["checkpoint_id"],
diff --git a/libs/scheduler-kafka/tests/test_subgraph_sync.py b/libs/scheduler-kafka/tests/test_subgraph_sync.py
index 50b33a39d..1071e0524 100644
--- a/libs/scheduler-kafka/tests/test_subgraph_sync.py
+++ b/libs/scheduler-kafka/tests/test_subgraph_sync.py
@@ -201,7 +201,7 @@ def test_subgraph_w_interrupt(
                                 "subgraph_counter": None,
                                 "call_counter": None,
                                 "interrupt_counter": None,
-                                "null_resume": None,
+                                "get_null_resume": None,
                                 "resume": [],
                             },
                             "checkpoint_id": None,
@@ -274,7 +274,7 @@ def test_subgraph_w_interrupt(
                                 "subgraph_counter": None,
                                 "call_counter": None,
                                 "interrupt_counter": None,
-                                "null_resume": None,
+                                "get_null_resume": None,
                                 "resume": [],
                             },
                             "checkpoint_id": c.config["configurable"]["checkpoint_id"],
@@ -377,7 +377,7 @@ def test_subgraph_w_interrupt(
                                 "subgraph_counter": None,
                                 "call_counter": None,
                                 "interrupt_counter": None,
-                                "null_resume": None,
+                                "get_null_resume": None,
                                 "resume": [],
                             },
                             "checkpoint_id": c.config["configurable"]["checkpoint_id"],
@@ -489,7 +489,7 @@ def test_subgraph_w_interrupt(
                                 "subgraph_counter": None,
                                 "call_counter": None,
                                 "interrupt_counter": None,
-                                "null_resume": None,
+                                "get_null_resume": None,
                                 "resume": [],
                             },
                             "checkpoint_id": None,
@@ -557,7 +557,7 @@ def test_subgraph_w_interrupt(
                                 "subgraph_counter": None,
                                 "call_counter": None,
                                 "interrupt_counter": None,
-                                "null_resume": None,
+                                "get_null_resume": None,
                                 "resume": [],
                             },
                             "checkpoint_id": c.config["configurable"]["checkpoint_id"],
@@ -681,7 +681,7 @@ def test_subgraph_w_interrupt(
                                 "subgraph_counter": None,
                                 "call_counter": None,
                                 "interrupt_counter": None,
-                                "null_resume": None,
+                                "get_null_resume": None,
                                 "resume": [],
                             },
                             "checkpoint_id": c.config["configurable"]["checkpoint_id"],