diff --git a/surfsense_backend/app/tasks/chat/stream_new_chat.py b/surfsense_backend/app/tasks/chat/stream_new_chat.py
index 0d72d8a2c..e33b67902 100644
--- a/surfsense_backend/app/tasks/chat/stream_new_chat.py
+++ b/surfsense_backend/app/tasks/chat/stream_new_chat.py
@@ -874,11 +874,12 @@ async def _stream_agent_events(
                 "create_google_drive_file",
                 "delete_google_drive_file",
                 "create_gmail_draft",
+                "update_gmail_draft",
                 "send_gmail_email",
                 "trash_gmail_email",
-                "create_google_calendar_event",
-                "update_google_calendar_event",
-                "delete_google_calendar_event",
+                "create_calendar_event",
+                "update_calendar_event",
+                "delete_calendar_event",
             ):
                 yield streaming_service.format_tool_output_available(
                     tool_call_id,