diff --git a/mix.exs b/mix.exs
index 19b773c..f453ac9 100644
--- a/mix.exs
+++ b/mix.exs
@@ -75,19 +75,17 @@ defmodule Mob.MixProject do
     </script>
     <script src="https://cdn.jsdelivr.net/npm/mermaid@11/dist/mermaid.min.js"></script>
     <script>
+      // Activate mermaid diagrams
       document.addEventListener("DOMContentLoaded", function () {
-        mermaid.initialize({ startOnLoad: false });
-        let id = 0;
-        for (const codeEl of document.querySelectorAll("pre code.mermaid")) {
-          const preEl = codeEl.closest("pre");
-          const graphEl = document.createElement("div");
-          const graphId = "mermaid-graph-" + id++;
-          mermaid.render(graphId, codeEl.textContent).then(({ svg, bindFunctions }) => {
-            graphEl.innerHTML = svg;
-            if (bindFunctions) bindFunctions(graphEl);
-            preEl.replaceWith(graphEl);
-          });
-        }
+        mermaid.initialize({ startOnLoad: false, theme: "dark" });
+        window.mermaid.init(undefined, document.querySelectorAll(".mermaid"));
+      });
+    </script>
+
+    <script>
+      // Ensure code blocks with language hints are highlighted
+      document.querySelectorAll("pre code").forEach(el => {
+        if (!el.className) el.className = "language-elixir";
       });
     </script>
     """