diff --git a/src/backend/langflow/components/agents/JsonAgent.py b/src/backend/langflow/components/agents/JsonAgent.py
index 935138142..71b276e26 100644
--- a/src/backend/langflow/components/agents/JsonAgent.py
+++ b/src/backend/langflow/components/agents/JsonAgent.py
@@ -4,8 +4,8 @@ from langchain.agents import AgentExecutor
 from typing import Callable
 from langflow.field_typing import (
     BaseLanguageModel,
-    BaseToolkit,
 )
+from langchain_community.agent_toolkits.base import BaseToolkit
 
 class JsonAgentComponent(CustomComponent):
     display_name = "JsonAgent"