diff --git a/src/crewai/memory/external/external_memory.py b/src/crewai/memory/external/external_memory.py
index 51ec8718e..0b2630f04 100644
--- a/src/crewai/memory/external/external_memory.py
+++ b/src/crewai/memory/external/external_memory.py
@@ -1,14 +1,16 @@
 import sys
-if sys.version_info < (3, 11):
-    from typing_extensions import Self
-    from typing import TYPE_CHECKING, Any, Dict, Optional
-else:
-    from typing import TYPE_CHECKING, Any, Dict, Optional, Self
 
 from crewai.memory.external.external_memory_item import ExternalMemoryItem
 from crewai.memory.memory import Memory
 from crewai.memory.storage.interface import Storage
 
+if sys.version_info < (3, 11):
+    from typing import TYPE_CHECKING, Any, Dict, Optional
+
+    from typing_extensions import Self
+else:
+    from typing import TYPE_CHECKING, Any, Dict, Optional, Self
+
 if TYPE_CHECKING:
     from crewai.memory.storage.mem0_storage import Mem0Storage
 
diff --git a/src/crewai/memory/memory.py b/src/crewai/memory/memory.py
index bd746b698..17e578808 100644
--- a/src/crewai/memory/memory.py
+++ b/src/crewai/memory/memory.py
@@ -1,12 +1,14 @@
 import sys
-if sys.version_info < (3, 11):
-    from typing_extensions import Self
-    from typing import Any, Dict, List, Optional
-else:
-    from typing import Any, Dict, List, Optional, Self
 
 from pydantic import BaseModel
 
+if sys.version_info < (3, 11):
+    from typing import Any, Dict, List, Optional
+
+    from typing_extensions import Self
+else:
+    from typing import Any, Dict, List, Optional, Self
+
 
 class Memory(BaseModel):
     """