diff --git a/src/backend/langflow/components/agents/ZeroShotAgent.py b/src/backend/langflow/components/agents/ZeroShotAgent.py
index fca29a9e0..bdc402059 100644
--- a/src/backend/langflow/components/agents/ZeroShotAgent.py
+++ b/src/backend/langflow/components/agents/ZeroShotAgent.py
@@ -1,10 +1,10 @@
 
 from langflow import CustomComponent
 from langchain.agents import ZeroShotAgent
+from langchain_core.tools import BaseTool
 from typing import List, Optional
 from langflow.field_typing import (
     BaseLanguageModel,
-    BaseTool,
 )
 
 class ZeroShotAgentComponent(CustomComponent):