diff --git a/langgraph/checkpoint/aiosqlite.py b/langgraph/checkpoint/aiosqlite.py
index 199a23c6c..a374886cd 100644
--- a/langgraph/checkpoint/aiosqlite.py
+++ b/langgraph/checkpoint/aiosqlite.py
@@ -1,11 +1,12 @@
 import pickle
 from contextlib import AbstractAsyncContextManager
 from types import TracebackType
-from typing import AsyncIterator, Optional, Self
+from typing import AsyncIterator, Optional
 
 import aiosqlite
 from langchain_core.pydantic_v1 import Field
 from langchain_core.runnables import RunnableConfig
+from typing_extensions import Self
 
 from langgraph.checkpoint.base import BaseCheckpointSaver, Checkpoint, CheckpointTuple
 
diff --git a/langgraph/checkpoint/sqlite.py b/langgraph/checkpoint/sqlite.py
index 6c8b75411..d2b801f19 100644
--- a/langgraph/checkpoint/sqlite.py
+++ b/langgraph/checkpoint/sqlite.py
@@ -2,10 +2,11 @@ import pickle
 import sqlite3
 from contextlib import AbstractContextManager, contextmanager
 from types import TracebackType
-from typing import AsyncIterator, Iterator, Optional, Self
+from typing import AsyncIterator, Iterator, Optional
 
 from langchain_core.pydantic_v1 import Field
 from langchain_core.runnables import RunnableConfig
+from typing_extensions import Self
 
 from langgraph.checkpoint.base import BaseCheckpointSaver, Checkpoint, CheckpointTuple