diff --git a/utils/langchain_helpers/oldschoolsearch.py b/utils/langchain_helpers/oldschoolsearch.py
index b91ec59..0fb5761 100644
--- a/utils/langchain_helpers/oldschoolsearch.py
+++ b/utils/langchain_helpers/oldschoolsearch.py
@@ -12,7 +12,7 @@
 
 from utils import openai_helpers
 from utils import redis_helpers
-
+from utils import cosmos_helpers
 
 DAVINCI_003_MODEL_MAX_TOKENS = int(os.environ["DAVINCI_003_MODEL_MAX_TOKENS"])
 ADA_002_MODEL_MAX_TOKENS     = int(os.environ["ADA_002_MODEL_MAX_TOKENS"])
@@ -92,4 +92,4 @@ def search(self, query, history, pre_context, topK=NUM_TOP_MATCHES):
 
         final_answer = openai_helpers.contact_openai(prompt, completion_model, MAX_OUTPUT_TOKENS)
 
-        return final_answer
\ No newline at end of file
+        return final_answer