diff --git a/llamaindex-mcp/ollama_client.ipynb b/llamaindex-mcp/ollama_client.ipynb
index db75f94e1..381919406 100644
--- a/llamaindex-mcp/ollama_client.ipynb
+++ b/llamaindex-mcp/ollama_client.ipynb
@@ -161,13 +161,12 @@
     "from llama_index.tools.mcp import McpToolSpec\n",
     "from llama_index.core.agent.workflow import FunctionAgent\n",
     "\n",
-    "async def get_agent(tools: McpToolSpec):\n",
-    "    tools = await tools.to_tool_list_async()\n",
+    "async def get_agent(tools: McpToolSpec, llm=Settings.llm):\n",
     "    agent = FunctionAgent(\n",
     "        name=\"Agent\",\n",
     "        description=\"An agent that can work with Our Database software.\",\n",
     "        tools=tools,\n",
-    "        llm=OpenAI(model=\"gpt-4\"),\n",
+    "        llm=llm,\n",
     "        system_prompt=SYSTEM_PROMPT,\n",
     "    )\n",
     "    return agent"