diff --git a/libs/vertexai/tests/unit_tests/test_function_utils.py b/libs/vertexai/tests/unit_tests/test_function_utils.py
index 96a04860..326c14e3 100644
--- a/libs/vertexai/tests/unit_tests/test_function_utils.py
+++ b/libs/vertexai/tests/unit_tests/test_function_utils.py
@@ -275,7 +275,7 @@ def search(question: str) -> str:
     parameters=gapic.Schema(
         type=gapic.Type.OBJECT,
         description="Search tool",
-        title="searchSchema",
+        title="search",
         properties={"question": gapic.Schema(type=gapic.Type.STRING, title="Question")},
         required=["question"],
     ),