diff --git a/libs/scheduler-kafka/tests/test_subgraph.py b/libs/scheduler-kafka/tests/test_subgraph.py
index 4ab92676c..1e1f1e396 100644
--- a/libs/scheduler-kafka/tests/test_subgraph.py
+++ b/libs/scheduler-kafka/tests/test_subgraph.py
@@ -191,6 +191,7 @@ async def test_subgraph_w_interrupt(
                             "__pregel_delegate": False,
                             "__pregel_read": None,
                             "__pregel_send": None,
+                            "__pregel_call": None,
                             "__pregel_ensure_latest": True,
                             "__pregel_dedupe_tasks": True,
                             "__pregel_resuming": False,
@@ -257,6 +258,7 @@ async def test_subgraph_w_interrupt(
                             "__pregel_delegate": False,
                             "__pregel_read": None,
                             "__pregel_send": None,
+                            "__pregel_call": None,
                             "__pregel_ensure_latest": True,
                             "__pregel_dedupe_tasks": True,
                             "__pregel_resuming": False,
@@ -353,6 +355,7 @@ async def test_subgraph_w_interrupt(
                             "__pregel_delegate": False,
                             "__pregel_read": None,
                             "__pregel_send": None,
+                            "__pregel_call": None,
                             "__pregel_ensure_latest": True,
                             "__pregel_dedupe_tasks": True,
                             "__pregel_resuming": False,
@@ -459,6 +462,7 @@ async def test_subgraph_w_interrupt(
                             "__pregel_delegate": False,
                             "__pregel_read": None,
                             "__pregel_send": None,
+                            "__pregel_call": None,
                             "__pregel_ensure_latest": True,
                             "__pregel_dedupe_tasks": True,
                             "__pregel_resuming": True,
@@ -520,6 +524,7 @@ async def test_subgraph_w_interrupt(
                             "__pregel_delegate": False,
                             "__pregel_read": None,
                             "__pregel_send": None,
+                            "__pregel_call": None,
                             "__pregel_ensure_latest": True,
                             "__pregel_dedupe_tasks": True,
                             "__pregel_resuming": True,
@@ -637,6 +642,7 @@ async def test_subgraph_w_interrupt(
                             "__pregel_delegate": False,
                             "__pregel_read": None,
                             "__pregel_send": None,
+                            "__pregel_call": None,
                             "__pregel_ensure_latest": True,
                             "__pregel_dedupe_tasks": True,
                             "__pregel_resuming": True,
diff --git a/libs/scheduler-kafka/tests/test_subgraph_sync.py b/libs/scheduler-kafka/tests/test_subgraph_sync.py
index 5fa43998a..210312b3b 100644
--- a/libs/scheduler-kafka/tests/test_subgraph_sync.py
+++ b/libs/scheduler-kafka/tests/test_subgraph_sync.py
@@ -190,6 +190,7 @@ def test_subgraph_w_interrupt(
                             "__pregel_delegate": False,
                             "__pregel_read": None,
                             "__pregel_send": None,
+                            "__pregel_call": None,
                             "__pregel_ensure_latest": True,
                             "__pregel_dedupe_tasks": True,
                             "__pregel_resuming": False,
@@ -256,6 +257,7 @@ def test_subgraph_w_interrupt(
                             "__pregel_delegate": False,
                             "__pregel_read": None,
                             "__pregel_send": None,
+                            "__pregel_call": None,
                             "__pregel_ensure_latest": True,
                             "__pregel_store": None,
                             "__pregel_dedupe_tasks": True,
@@ -352,6 +354,7 @@ def test_subgraph_w_interrupt(
                             "__pregel_delegate": False,
                             "__pregel_read": None,
                             "__pregel_send": None,
+                            "__pregel_call": None,
                             "__pregel_ensure_latest": True,
                             "__pregel_dedupe_tasks": True,
                             "__pregel_store": None,
@@ -457,6 +460,7 @@ def test_subgraph_w_interrupt(
                             "__pregel_delegate": False,
                             "__pregel_read": None,
                             "__pregel_send": None,
+                            "__pregel_call": None,
                             "__pregel_ensure_latest": True,
                             "__pregel_dedupe_tasks": True,
                             "__pregel_store": None,
@@ -518,6 +522,7 @@ def test_subgraph_w_interrupt(
                             "__pregel_delegate": False,
                             "__pregel_read": None,
                             "__pregel_send": None,
+                            "__pregel_call": None,
                             "__pregel_ensure_latest": True,
                             "__pregel_dedupe_tasks": True,
                             "__pregel_store": None,
@@ -635,6 +640,7 @@ def test_subgraph_w_interrupt(
                             "__pregel_delegate": False,
                             "__pregel_read": None,
                             "__pregel_send": None,
+                            "__pregel_call": None,
                             "__pregel_ensure_latest": True,
                             "__pregel_dedupe_tasks": True,
                             "__pregel_resuming": True,