diff --git a/index.html b/index.html
index 466e0f7..7174939 100644
--- a/index.html
+++ b/index.html
@@ -431,7 +431,10 @@
       };
       const $elements = document.getElementById('elements');
       const updateStyle = (isDark) => {
-        for (const pre of $elements.querySelectorAll('pre.sl-code-viewer')) {
+        for (const pre of [
+          ...Array.from($elements.querySelectorAll('pre.sl-code-viewer')),
+          ...Array.from($elements.querySelectorAll('div.sl-code-editor')),
+        ]) {
           const isPreDark = isDark || !!pre.closest('.sl-inverted');
           const theme = isPreDark ? darkTheme : lightTheme;
           for (const span of pre.querySelectorAll('span')) {