diff --git a/packages/stage-ui/src/components/scenarios/chat/components/error-item.vue b/packages/stage-ui/src/components/scenarios/chat/components/error-item.vue
index ce8464e0a..47b72f2b9 100644
--- a/packages/stage-ui/src/components/scenarios/chat/components/error-item.vue
+++ b/packages/stage-ui/src/components/scenarios/chat/components/error-item.vue
@@ -83,7 +83,7 @@ const copyText = computed(() => getChatHistoryItemCopyText(props.message as Chat
     <div
       v-if="canRetry && !showPlaceholder"
       :class="[
-        'self-end mt-1',
+        'self-end mt-1 w-fit',
       ]"
     >
       <Button
diff --git a/packages/stage-ui/src/components/scenarios/chat/components/response-part.vue b/packages/stage-ui/src/components/scenarios/chat/components/response-part.vue
index a5d809fc8..e96880e45 100644
--- a/packages/stage-ui/src/components/scenarios/chat/components/response-part.vue
+++ b/packages/stage-ui/src/components/scenarios/chat/components/response-part.vue
@@ -15,7 +15,6 @@ const reasoningContent = computed(() => props.message.categorization?.reasoning?
 const hasReasoning = computed(() => reasoningContent.value.length > 0)
 
 const containerClasses = computed(() => [
-  'mt-2',
   props.variant === 'mobile' ? 'text-xs' : 'text-sm',
 ])
 </script>