refactor: method of setting context menu

2 files changed, 2 insertions, 4 deletions

diff --git a/src/dumbymap.mjs b/src/dumbymap.mjs
index 7917499..a228765 100644
--- a/src/dumbymap.mjs
+++ b/src/dumbymap.mjs
@@ -140,6 +140,8 @@ export const generateMaps = (container, { delay } = {}) => {
         container.oncontextmenu = (e) => {
           const menu = originalCallback(e)
           menuCallback(e, menu)
+          menu.style.transform = ''
+          shiftByWindow(menu)
         }
       },
       focusNextMap: throttle(utils.focusNextMap, utils.focusDelay),
diff --git a/src/editor.mjs b/src/editor.mjs
index 4a58f12..d1f84a2 100644
--- a/src/editor.mjs
+++ b/src/editor.mjs
@@ -444,10 +444,6 @@ const menuForEditor = (event, menu) => {
     })
     menu.insertBefore(item, menu.firstChild)
   }
-
-  // Prevent menu appears outside of window
-  menu.style.transform = ''
-  shiftByWindow(menu)
 }
 
 /**