diff --git a/swarms/memory/base.py b/swarms/memory/base.py
index 6af5015b..ff51e0da 100644
--- a/swarms/memory/base.py
+++ b/swarms/memory/base.py
@@ -26,7 +26,7 @@ from langchain.callbacks.manager import (
     CallbackManagerForRetrieverRun,
 )
 
-from swarms.agents.memory.document import Document
+from swarms.memory.document import Document
 from swarms.utils.embeddings.base import Embeddings
 
 from langchain.schema import BaseRetriever
diff --git a/swarms/memory/utils.py b/swarms/memory/utils.py
index 5c4792e6..43f0622e 100644
--- a/swarms/memory/utils.py
+++ b/swarms/memory/utils.py
@@ -1,6 +1,6 @@
 from typing import Any, Dict, List
 
-from swarms.agents.memory.base import get_buffer_string
+from swarms.memory.base import get_buffer_string
 
 
 
diff --git a/swarms/orchestrator/orchestrate.py b/swarms/orchestrator/orchestrate.py
index 902b5f4b..4e697598 100644
--- a/swarms/orchestrator/orchestrate.py
+++ b/swarms/orchestrator/orchestrate.py
@@ -3,7 +3,7 @@ import threading
 from abc import ABC, abstractmethod
 from typing import Any, Dict, List
 
-from swarms.agents.memory.ocean import OceanDB
+from swarms.memory.ocean import OceanDB
 
 
 class Orchestrator(ABC):
diff --git a/tests/agents/memory/main.py b/tests/agents/memory/main.py
index f7ae0ec4..c156ab3d 100644
--- a/tests/agents/memory/main.py
+++ b/tests/agents/memory/main.py
@@ -1,6 +1,6 @@
 import pytest
 from unittest.mock import Mock
-from swarms.agents.memory.oceandb import OceanDB
+from swarms.memory.oceandb import OceanDB
 
 
 @pytest.fixture