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