diff --git a/src/components/ChatUI/CodeHighlight.tsx b/src/components/ChatUI/CodeHighlight.tsx
index 2d128bb0..1cef7c7d 100644
--- a/src/components/ChatUI/CodeHighlight.tsx
+++ b/src/components/ChatUI/CodeHighlight.tsx
@@ -42,7 +42,7 @@ export const CodeHighlight = ({
   }, [code]);
 
   return !isInline ? (
-    <div className='relative'>
+    <div className='shiki not-prose relative [&_pre]:overflow-auto [&_pre]:rounded-lg [&_pre]:px-6 [&_pre]:py-5'>
       {language ? (
         <span className='absolute right-3 top-2 text-xs tracking-tighter text-muted-foreground/85'>
           {language}