diff --git a/src/components/ai-elements/message.tsx b/src/components/ai-elements/message.tsx
index ac6dcb8..dc56af7 100644
--- a/src/components/ai-elements/message.tsx
+++ b/src/components/ai-elements/message.tsx
@@ -34,7 +34,7 @@ export const MessageContent = ({
className={cn(
'flex flex-col gap-2 overflow-hidden rounded-lg px-4 py-3 text-foreground text-sm',
'group-[.is-user]:bg-primary group-[.is-user]:text-primary-foreground',
- 'group-[.is-assistant]:bg-secondary group-[.is-assistant]:text-foreground',
+ 'group-[.is-assistant]:bg-card group-[.is-assistant]:text-card-foreground',
'is-user:dark',
className
)}