diff --git a/libs/checkpoint-postgres/tests/test_async.py b/libs/checkpoint-postgres/tests/test_async.py
index 4a9f08336..9dc5c78cd 100644
--- a/libs/checkpoint-postgres/tests/test_async.py
+++ b/libs/checkpoint-postgres/tests/test_async.py
@@ -361,9 +361,9 @@ async def test_get_checkpoint_no_channel_values(
 
         load_checkpoint_tuple = saver._load_checkpoint_tuple
 
-        def patched_load_checkpoint_tuple(value):
+        async def patched_load_checkpoint_tuple(value, cur):
             value["checkpoint"].pop("channel_values", None)
-            return load_checkpoint_tuple(value)
+            return await load_checkpoint_tuple(value, cur)
 
         monkeypatch.setattr(
             saver, "_load_checkpoint_tuple", patched_load_checkpoint_tuple
diff --git a/libs/checkpoint-postgres/tests/test_sync.py b/libs/checkpoint-postgres/tests/test_sync.py
index 63b93715f..cc78c615e 100644
--- a/libs/checkpoint-postgres/tests/test_sync.py
+++ b/libs/checkpoint-postgres/tests/test_sync.py
@@ -348,9 +348,9 @@ def test_get_checkpoint_no_channel_values(
 
         load_checkpoint_tuple = saver._load_checkpoint_tuple
 
-        def patched_load_checkpoint_tuple(value):
+        def patched_load_checkpoint_tuple(value, cur):
             value["checkpoint"].pop("channel_values", None)
-            return load_checkpoint_tuple(value)
+            return load_checkpoint_tuple(value, cur)
 
         monkeypatch.setattr(
             saver, "_load_checkpoint_tuple", patched_load_checkpoint_tuple
diff --git a/libs/checkpoint/langgraph/checkpoint/base/__init__.py b/libs/checkpoint/langgraph/checkpoint/base/__init__.py
index 10382519e..f8fb8bbb7 100644
--- a/libs/checkpoint/langgraph/checkpoint/base/__init__.py
+++ b/libs/checkpoint/langgraph/checkpoint/base/__init__.py
@@ -516,7 +516,7 @@ class BaseCheckpointSaver(Generic[V]):
         self, config: RunnableConfig, channel: str
     ) -> List[Any]:  # noqa: UP006
         """Async version of get_channel_writes."""
-        from langgraph.types import Overwrite  # type: ignore[import-untyped]
+        from langgraph.types import Overwrite
 
         if getattr(_DELTA_RECONSTRUCTION, "active", False):
             return []