diff --git a/tests/unittests/tools/application_integration_tool/test_application_integration_toolset.py b/tests/unittests/tools/application_integration_tool/test_application_integration_toolset.py
index a95e29f..c642441 100644
--- a/tests/unittests/tools/application_integration_tool/test_application_integration_toolset.py
+++ b/tests/unittests/tools/application_integration_tool/test_application_integration_toolset.py
@@ -532,7 +532,7 @@ def test_init_with_connection_and_custom_auth(
       location,
       connection=connection_name,
       actions=actions_list,
-      tool_name=tool_name,
+      tool_name_prefix=tool_name,
       tool_instructions=tool_instructions,
       auth_scheme=oauth2_scheme,
       auth_credential=auth_credential,
@@ -600,7 +600,7 @@ def test_init_with_connection_with_auth_override_disabled_and_custom_auth(
       location,
       connection=connection_name,
       actions=actions_list,
-      tool_name=tool_name,
+      tool_name_prefix=tool_name,
       tool_instructions=tool_instructions,
       auth_scheme=oauth2_scheme,
       auth_credential=auth_credential,