diff --git a/src/web/features/chat/parts/CodeBlock.tsx b/src/web/features/chat/parts/CodeBlock.tsx
index 6895b8c..2558955 100644
--- a/src/web/features/chat/parts/CodeBlock.tsx
+++ b/src/web/features/chat/parts/CodeBlock.tsx
@@ -63,9 +63,11 @@ export function CodeBlock({ children, className: _className, isStreaming }: Code
       {highlighted ? (
         <div className="code-block-body" dangerouslySetInnerHTML={{ __html: highlighted }} />
       ) : (
-        <pre className="code-block-body">
-          <code>{codeText}</code>
-        </pre>
+        <div className="code-block-body">
+          <pre>
+            <code>{codeText}</code>
+          </pre>
+        </div>
       )}
     </div>
   );
diff --git a/src/web/styles.css b/src/web/styles.css
index 50c3d97..6ed9c9c 100644
--- a/src/web/styles.css
+++ b/src/web/styles.css
@@ -408,9 +408,12 @@ body {
 
 .code-block-body {
   margin: 0;
+}
+
+.code-block-body > pre {
   padding: var(--ant-padding-sm);
+  margin: 0;
   overflow-x: auto;
-  background: var(--ant-color-bg-container);
 }
 
 .code-block-body code {