diff --git a/tests/pass_through_unit_tests/test_unit_test_passthrough_router.py b/tests/pass_through_unit_tests/test_unit_test_passthrough_router.py index 3f62f557915..8e016b68d05 100644 --- a/tests/pass_through_unit_tests/test_unit_test_passthrough_router.py +++ b/tests/pass_through_unit_tests/test_unit_test_passthrough_router.py @@ -102,7 +102,7 @@ class TestPassthroughEndpointRouter(unittest.TestCase): mock_get_secret.return_value = "env_azure_key" result = self.router.get_credentials("azure", None) self.assertEqual(result, "env_azure_key") - mock_get_secret.assert_called_once_with("AZURE_AI_API_KEY") + mock_get_secret.assert_called_once_with("AZURE_API_KEY") def test_default_env_variable_method(self): """