1 files changed, 2 insertions, 2 deletions

diff --git a/src/editor.mjs b/src/editor.mjs
index a7bdd77..20bcc2a 100644
--- a/src/editor.mjs
+++ b/src/editor.mjs
@@ -12,12 +12,12 @@ let dumbymap;
 
 new MutationObserver(() => {
   if (
-    document.body.getAttribute('data-mode') === 'editing' &&
+    document.querySelector('[data-mode]').getAttribute('data-mode') === 'editing' &&
     HtmlContainer.getAttribute('data-layout') !== 'normal'
   ) {
     HtmlContainer.setAttribute('data-layout', 'normal');
   }
-}).observe(document.body, {
+}).observe(document.querySelector('[data-mode]'), {
   attributes: true,
   attributeFilter: ['data-mode'],
 });