diff --git a/tests/test_litellm/proxy/hooks/test_key_management_event_hooks.py b/tests/test_litellm/proxy/hooks/test_key_management_event_hooks.py
index 011031c1e4f..97c1733a935 100644
--- a/tests/test_litellm/proxy/hooks/test_key_management_event_hooks.py
+++ b/tests/test_litellm/proxy/hooks/test_key_management_event_hooks.py
@@ -40,6 +40,7 @@ class TestKeyManagementEventHooksIndependentOperations:
         mock_data = MagicMock()
         mock_data.key_alias = "test-key-alias"
         mock_data.team_id = None
+        mock_data.send_invite_email = True
 
         mock_response = MagicMock()
         mock_response.model_dump.return_value = {"key": "sk-test", "token": "test-token"}
@@ -59,6 +60,10 @@ class TestKeyManagementEventHooksIndependentOperations:
             KeyManagementEventHooks,
             "_store_virtual_key_in_secret_manager",
             side_effect=mock_store_secret,
+        ), patch.object(
+            KeyManagementEventHooks,
+            "_is_email_sending_enabled",
+            return_value=True,
         ), patch(
             "litellm.store_audit_logs", False
         ), patch(
@@ -96,6 +101,7 @@ class TestKeyManagementEventHooksIndependentOperations:
         mock_data = MagicMock()
         mock_data.key_alias = "test-key-alias"
         mock_data.team_id = None
+        mock_data.send_invite_email = True
 
         mock_response = MagicMock()
         mock_response.model_dump.return_value = {"key": "sk-test", "token": "test-token"}
@@ -115,6 +121,10 @@ class TestKeyManagementEventHooksIndependentOperations:
             KeyManagementEventHooks,
             "_store_virtual_key_in_secret_manager",
             side_effect=mock_store_secret_raises,
+        ), patch.object(
+            KeyManagementEventHooks,
+            "_is_email_sending_enabled",
+            return_value=True,
         ), patch(
             "litellm.store_audit_logs", False
         ), patch(