diff --git a/libs/langgraph/tests/test_pregel.py b/libs/langgraph/tests/test_pregel.py
index 348922e2b..a17cf4e99 100644
--- a/libs/langgraph/tests/test_pregel.py
+++ b/libs/langgraph/tests/test_pregel.py
@@ -11917,6 +11917,9 @@ def test_debug_nested_subgraphs():
         clean_config["thread_id"] = config["configurable"]["thread_id"]
         clean_config["checkpoint_id"] = config["configurable"]["checkpoint_id"]
         clean_config["checkpoint_ns"] = config["configurable"]["checkpoint_ns"]
+        clean_config["checkpoint_map"] = config["configurable"]["checkpoint_map"]
+        if "checkpoint_map" in config["configurable"]:
+            clean_config["checkpoint_map"] = config["configurable"]["checkpoint_map"]
 
         return clean_config
 
diff --git a/libs/langgraph/tests/test_pregel_async.py b/libs/langgraph/tests/test_pregel_async.py
index 96317d041..928cfc5eb 100644
--- a/libs/langgraph/tests/test_pregel_async.py
+++ b/libs/langgraph/tests/test_pregel_async.py
@@ -10143,6 +10143,8 @@ async def test_debug_nested_subgraphs():
         clean_config["thread_id"] = config["configurable"]["thread_id"]
         clean_config["checkpoint_id"] = config["configurable"]["checkpoint_id"]
         clean_config["checkpoint_ns"] = config["configurable"]["checkpoint_ns"]
+        if "checkpoint_map" in config["configurable"]:
+            clean_config["checkpoint_map"] = config["configurable"]["checkpoint_map"]
 
         return clean_config