diff --git a/ra_aid/tools/memory.py b/ra_aid/tools/memory.py
index 69dfdd4..e5262e9 100644
--- a/ra_aid/tools/memory.py
+++ b/ra_aid/tools/memory.py
@@ -1,4 +1,5 @@
-from typing import Dict, List, Any, Union, TypedDict, Optional, Set
+from typing import Dict, List, Any, Union, Optional, Set
+from typing_extensions import TypedDict
 
 class WorkLogEntry(TypedDict):
     timestamp: str