diff --git a/src/backend/base/langflow/components/vectorstores/MongoDBAtlasVector.py b/src/backend/base/langflow/components/vectorstores/MongoDBAtlasVector.py
index ca6bacfc7..e7c5c16bc 100644
--- a/src/backend/base/langflow/components/vectorstores/MongoDBAtlasVector.py
+++ b/src/backend/base/langflow/components/vectorstores/MongoDBAtlasVector.py
@@ -83,7 +83,7 @@ class MongoVectorStoreComponent(LCVectorStoreComponent):
         return vector_store
 
     def search_documents(self) -> List[Data]:
-        from typing import Union
+        from typing import Optional, Union
         from bson import ObjectId
         from langchain.schema import Document
         import json