diff --git a/scripts/mdbook-shiki.ts b/scripts/mdbook-shiki.ts
index 5be73f86eb..e45f6b8239 100755
--- a/scripts/mdbook-shiki.ts
+++ b/scripts/mdbook-shiki.ts
@@ -12,7 +12,7 @@ import {
   extractShikiLines,
 } from "../src/utils/highlighting/shiki-shared";
 import { renderToStaticMarkup } from "react-dom/server";
-import { CodeBlockSSR } from "../src/components/Messages/CodeBlockSSR";
+import { CodeBlockSSR } from "../src/browser/components/Messages/CodeBlockSSR";
 
 interface Chapter {
   name: string;
@@ -92,7 +92,7 @@ async function processMarkdown(
 
       const html = highlighter.codeToHtml(trimmedCode, {
         lang: shikiLang,
-        theme: SHIKI_THEME,
+        theme: SHIKI_DARK_THEME,
       });
 
       const gridHtml = generateGridHtml(html, trimmedCode);
@@ -156,7 +156,7 @@ async function main() {
 
   // Initialize Shiki highlighter
   const highlighter = await createHighlighter({
-    themes: [SHIKI_THEME],
+    themes: [SHIKI_DARK_THEME],
     langs: [], // Load on-demand
   });