1 files changed, 4 insertions, 1 deletions

diff --git a/src/MenuItem.mjs b/src/MenuItem.mjs
index f4cd518..74138ec 100644
--- a/src/MenuItem.mjs
+++ b/src/MenuItem.mjs
@@ -89,7 +89,10 @@ export const pickLayoutItem = dumbymap =>
     items: [
       new Item({
         text: 'EDIT',
-        onclick: () => document.body.setAttribute('data-mode', 'editing'),
+        onclick: () =>
+          dumbymap.container
+            .closest('.playground')
+            .setAttribute('data-mode', 'editing'),
       }).element,
       ...dumbymap.layouts.map(
         layout =>