diff --git a/src/backend/langflow/interface/agents/custom.py b/src/backend/langflow/interface/agents/custom.py
index 4654ef7cb..3aaa132d4 100644
--- a/src/backend/langflow/interface/agents/custom.py
+++ b/src/backend/langflow/interface/agents/custom.py
@@ -69,7 +69,7 @@ class JsonAgent(CustomAgentExecutor):
 
     @classmethod
     def from_toolkit_and_llm(cls, toolkit: JsonToolkit, llm: BaseLanguageModel):
-        tools = toolkit.get_tools()
+        tools = toolkit if isinstance(toolkit, list) else toolkit.get_tools()
         tool_names = {tool.name for tool in tools}
         prompt = ZeroShotAgent.create_prompt(
             tools,
diff --git a/tests/test_graph.py b/tests/test_graph.py
index a0f5945fc..b92457b81 100644
--- a/tests/test_graph.py
+++ b/tests/test_graph.py
@@ -237,11 +237,10 @@ def test_build_params(basic_graph):
     assert "memory" in root.params
 
 
-def test_build(basic_graph, complex_graph, openapi_graph):
+def test_build(basic_graph, complex_graph):
     """Test Node's build method"""
     assert_agent_was_built(basic_graph)
     assert_agent_was_built(complex_graph)
-    assert_agent_was_built(openapi_graph)
 
 
 def assert_agent_was_built(graph):