diff --git a/backend/onyx/chat/process_message.py b/backend/onyx/chat/process_message.py
index fec9c123f8d..f4e06cf28b7 100644
--- a/backend/onyx/chat/process_message.py
+++ b/backend/onyx/chat/process_message.py
@@ -51,7 +51,7 @@ from onyx.context.search.enums import SearchType
 from onyx.context.search.models import InferenceSection
 from onyx.context.search.models import RetrievalDetails
 from onyx.context.search.models import SearchRequest
-from onyx.context.search.retrieve_orig_question_documents.search_runner import (
+from onyx.context.search.retrieval.search_runner import (
     inference_sections_from_ids,
 )
 from onyx.context.search.utils import chunks_or_sections_to_search_docs
diff --git a/backend/onyx/context/search/pipeline.py b/backend/onyx/context/search/pipeline.py
index ea6c458c70e..eace5b799e7 100644
--- a/backend/onyx/context/search/pipeline.py
+++ b/backend/onyx/context/search/pipeline.py
@@ -24,7 +24,7 @@ from onyx.context.search.models import SearchRequest
 from onyx.context.search.postprocessing.postprocessing import cleanup_chunks
 from onyx.context.search.postprocessing.postprocessing import search_postprocessing
 from onyx.context.search.preprocessing.preprocessing import retrieval_preprocessing
-from onyx.context.search.retrieve_orig_question_documents.search_runner import (
+from onyx.context.search.retrieval.search_runner import (
     retrieve_chunks,
 )
 from onyx.context.search.utils import inference_section_from_chunks
diff --git a/backend/onyx/context/search/preprocessing/preprocessing.py b/backend/onyx/context/search/preprocessing/preprocessing.py
index 91b2843fc02..da228f5f1fb 100644
--- a/backend/onyx/context/search/preprocessing/preprocessing.py
+++ b/backend/onyx/context/search/preprocessing/preprocessing.py
@@ -20,7 +20,7 @@ from onyx.context.search.models import SearchRequest
 from onyx.context.search.preprocessing.access_filters import (
     build_access_filters_for_user,
 )
-from onyx.context.search.retrieve_orig_question_documents.search_runner import (
+from onyx.context.search.retrieval.search_runner import (
     remove_stop_words_and_punctuation,
 )
 from onyx.db.engine import CURRENT_TENANT_ID_CONTEXTVAR
diff --git a/backend/onyx/setup.py b/backend/onyx/setup.py
index c37e47b3f73..9919c48c5a9 100644
--- a/backend/onyx/setup.py
+++ b/backend/onyx/setup.py
@@ -11,7 +11,7 @@ from onyx.configs.model_configs import FAST_GEN_AI_MODEL_VERSION
 from onyx.configs.model_configs import GEN_AI_API_KEY
 from onyx.configs.model_configs import GEN_AI_MODEL_VERSION
 from onyx.context.search.models import SavedSearchSettings
-from onyx.context.search.retrieve_orig_question_documents.search_runner import (
+from onyx.context.search.retrieval.search_runner import (
     download_nltk_data,
 )
 from onyx.db.connector import check_connectors_exist