diff --git a/firebase-ai/src/androidTest/kotlin/com/google/firebase/ai/ToolTests.kt b/firebase-ai/src/androidTest/kotlin/com/google/firebase/ai/ToolTests.kt
index 92ebbf4fc97..5aef0f429c6 100644
--- a/firebase-ai/src/androidTest/kotlin/com/google/firebase/ai/ToolTests.kt
+++ b/firebase-ai/src/androidTest/kotlin/com/google/firebase/ai/ToolTests.kt
@@ -304,7 +304,7 @@ class ToolTests {
               ToolConfig(
                 functionCallingConfig = FunctionCallingConfig(FunctionCallingConfig.Mode.ANY)
               ),
-            tools = listOf(Tool(functions.toList())),
+            tools = listOf(Tool.functionDeclarations(functions.toList())),
           )
       return model
     }