diff --git a/src/backend/langflow/components/agents/VectorStoreRouterAgent.py b/src/backend/langflow/components/agents/VectorStoreRouterAgent.py
index 6aec821c8..d613f3080 100644
--- a/src/backend/langflow/components/agents/VectorStoreRouterAgent.py
+++ b/src/backend/langflow/components/agents/VectorStoreRouterAgent.py
@@ -1,7 +1,7 @@
 
 from langflow import CustomComponent
-from langchain.llms import BaseLanguageModel
-from langchain.vectorstores import VectorStoreRouterToolkit
+from langchain_core.language_models.base import BaseLanguageModel
+from langchain.agents.agent_toolkits.vectorstore.toolkit import VectorStoreRouterToolkit
 from langchain.agents import AgentExecutor
 from typing import Callable