diff --git a/src/intelligence_layer/connectors/__init__.py b/src/intelligence_layer/connectors/__init__.py
index 492184e98..0f199d3dd 100644
--- a/src/intelligence_layer/connectors/__init__.py
+++ b/src/intelligence_layer/connectors/__init__.py
@@ -1,4 +1,4 @@
-from .document_index import DocumentIndex
+from .document_index.document_index import DocumentIndex
 from .retrievers.base_retriever import BaseRetriever, Document, SearchResult
 from .retrievers.document_index_retriever import DocumentIndexRetriever
 from .retrievers.qdrant_in_memory_retriever import (
diff --git a/src/intelligence_layer/connectors/document_index.py b/src/intelligence_layer/connectors/document_index/document_index.py
similarity index 100%
rename from src/intelligence_layer/connectors/document_index.py
rename to src/intelligence_layer/connectors/document_index/document_index.py
diff --git a/src/intelligence_layer/connectors/retrievers/document_index_retriever.py b/src/intelligence_layer/connectors/retrievers/document_index_retriever.py
index 306856071..319b40974 100644
--- a/src/intelligence_layer/connectors/retrievers/document_index_retriever.py
+++ b/src/intelligence_layer/connectors/retrievers/document_index_retriever.py
@@ -1,6 +1,6 @@
 from typing import Sequence
 
-from intelligence_layer.connectors.document_index import DocumentIndex
+from intelligence_layer.connectors.document_index.document_index import DocumentIndex
 from intelligence_layer.connectors.retrievers.base_retriever import (
     BaseRetriever,
     Document,
diff --git a/tests/conftest.py b/tests/conftest.py
index e811937db..64657a67a 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -10,7 +10,7 @@
 from intelligence_layer.connectors.retrievers.document_index_retriever import (
     DocumentIndexRetriever,
 )
-from intelligence_layer.connectors.document_index import DocumentIndex
+from intelligence_layer.connectors.document_index.document_index import DocumentIndex
 from intelligence_layer.connectors.retrievers.qdrant_in_memory_retriever import (
     QdrantInMemoryRetriever,
     RetrieverType,