diff --git a/src/neo4j_graphrag/types.py b/src/neo4j_graphrag/types.py
index 1c0b74542..d6b969a51 100644
--- a/src/neo4j_graphrag/types.py
+++ b/src/neo4j_graphrag/types.py
@@ -16,7 +16,8 @@
 
 import warnings
 from enum import Enum
-from typing import Any, Callable, Literal, Optional, TypedDict, Union
+from typing import Any, Callable, Literal, Optional, Union
+from typing_extensions import TypedDict
 
 import neo4j
 from pydantic import (