diff --git a/src/backend/base/langflow/components/vectorstores/Redis.py b/src/backend/base/langflow/components/vectorstores/Redis.py
index 65b1b6f70..a7d6ed7b9 100644
--- a/src/backend/base/langflow/components/vectorstores/Redis.py
+++ b/src/backend/base/langflow/components/vectorstores/Redis.py
@@ -1,11 +1,10 @@
-from typing import Optional, Union
+from typing import Optional
 
 from langchain_community.vectorstores.redis import Redis
 from langchain_core.embeddings import Embeddings
-from langchain_core.retrievers import BaseRetriever
-from langchain_core.vectorstores import VectorStore
 
 from langflow.custom import CustomComponent
+from langflow.field_typing import Retriever
 from langflow.schema import Data
 
 
@@ -45,7 +44,7 @@ class RedisVectorStoreComponent(CustomComponent):
         redis_index_name: str,
         schema: Optional[str] = None,
         inputs: Optional[Data] = None,
-    ) -> Union[VectorStore, BaseRetriever]:
+    ) -> Retriever:
         """
         Builds the Vector Store or BaseRetriever object.
 
@@ -82,3 +81,4 @@ class RedisVectorStoreComponent(CustomComponent):
                 index_name=redis_index_name,
             )
         return redis_vs
+        return redis_vs