diff --git a/templates/gemini-functions-agent/gemini_functions_agent/agent.py b/templates/gemini-functions-agent/gemini_functions_agent/agent.py
index 1f1c756d7c4..38ffc315ee9 100644
--- a/templates/gemini-functions-agent/gemini_functions_agent/agent.py
+++ b/templates/gemini-functions-agent/gemini_functions_agent/agent.py
@@ -32,7 +32,7 @@ prompt = ChatPromptTemplate.from_messages(
     ]
 )
 
-llm_with_tools = llm.bind(functions=[tavily_tool])
+llm_with_tools = llm.bind(functions=tools)
 
 
 def _format_chat_history(chat_history: List[Tuple[str, str]]):