diff --git a/src/crewai/memory/memory.py b/src/crewai/memory/memory.py
index 059adf744..788fa898a 100644
--- a/src/crewai/memory/memory.py
+++ b/src/crewai/memory/memory.py
@@ -1,4 +1,6 @@
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List, Optional, Union
+
+from pydantic import BaseModel
 
 from pydantic import BaseModel