diff --git a/app/javascript/dashboard/components/widgets/WootWriter/CopilotMenuBar.vue b/app/javascript/dashboard/components/widgets/WootWriter/CopilotMenuBar.vue
index 772602c2a..d91b5bd5c 100644
--- a/app/javascript/dashboard/components/widgets/WootWriter/CopilotMenuBar.vue
+++ b/app/javascript/dashboard/components/widgets/WootWriter/CopilotMenuBar.vue
@@ -33,7 +33,7 @@ const menuItems = computed(() => {
       label: t(
         'INTEGRATION_SETTINGS.OPEN_AI.REPLY_OPTIONS.IMPROVE_REPLY_SELECTION'
       ),
-      key: 'rephrase_selection',
+      key: 'improve_selection',
       icon: 'i-fluent-pen-sparkle-24-regular',
     });
   } else if (
@@ -42,7 +42,7 @@ const menuItems = computed(() => {
   ) {
     items.push({
       label: t('INTEGRATION_SETTINGS.OPEN_AI.REPLY_OPTIONS.IMPROVE_REPLY'),
-      key: 'rephrase',
+      key: 'improve',
       icon: 'i-fluent-pen-sparkle-24-regular',
     });
   }
diff --git a/enterprise/lib/enterprise/integrations/openai_processor_service.rb b/enterprise/lib/enterprise/integrations/openai_processor_service.rb
index 57639e8cb..2a241d7a3 100644
--- a/enterprise/lib/enterprise/integrations/openai_processor_service.rb
+++ b/enterprise/lib/enterprise/integrations/openai_processor_service.rb
@@ -1,6 +1,6 @@
 module Enterprise::Integrations::OpenaiProcessorService
   ALLOWED_EVENT_NAMES = %w[summarize reply_suggestion label_suggestion fix_spelling_grammar
-                           friendly casual professional confident straightforward].freeze
+                           friendly casual professional confident straightforward improve].freeze
   CACHEABLE_EVENTS = %w[label_suggestion].freeze
 
   def label_suggestion_message
diff --git a/lib/integrations/llm_base_service.rb b/lib/integrations/llm_base_service.rb
index accca98b4..397888b83 100644
--- a/lib/integrations/llm_base_service.rb
+++ b/lib/integrations/llm_base_service.rb
@@ -8,7 +8,7 @@ class Integrations::LlmBaseService
   TOKEN_LIMIT = 400_000
   GPT_MODEL = Llm::Config::DEFAULT_MODEL
   ALLOWED_EVENT_NAMES = %w[summarize reply_suggestion fix_spelling_grammar casual professional friendly confident
-                           straightforward].freeze
+                           straightforward improve].freeze
   CACHEABLE_EVENTS = %w[].freeze
 
   pattr_initialize [:hook!, :event!]