diff --git a/themes/mermaid_loader.js b/themes/mermaid_loader.js
index 183cbc3..64fe4f8 100644
--- a/themes/mermaid_loader.js
+++ b/themes/mermaid_loader.js
@@ -102,6 +102,10 @@ const uml = async className => {
             messageFontSize: "16px"
         }
     }
+    if (document.body.classList.contains("dark")) {
+        defaultConfig.theme = "dark"
+    }
+    
     const Module = await import('./file=themes/mermaid_editor.js');
 
     function do_render(block, code, codeContent) {
diff --git a/themes/theme.py b/themes/theme.py
index 8e8d86a..ff49066 100644
--- a/themes/theme.py
+++ b/themes/theme.py
@@ -111,6 +111,7 @@ js_code_for_toggle_darkmode = """() => {
     } else {
         document.querySelector('body').classList.add('dark');
     }
+    document.querySelectorAll('code2').forEach(code => {code.remove();})
 }"""