From 35b533045ca28cb459093554e2a0daf60076347c Mon Sep 17 00:00:00 2001
From: Alexander Zinchuk <alx.zinchuk@gmail.com>
Date: Tue, 14 Dec 2021 22:41:03 +0400
Subject: [PATCH] [Perf] Message / Context Menu: Speed up rendering (#1588)

---
 src/hooks/useContextMenuHandlers.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/hooks/useContextMenuHandlers.ts b/src/hooks/useContextMenuHandlers.ts
index acdbc286a..4150c5e06 100644
--- a/src/hooks/useContextMenuHandlers.ts
+++ b/src/hooks/useContextMenuHandlers.ts
@@ -30,12 +30,12 @@ export default (
 
   const handleBeforeContextMenu = useCallback((e: React.MouseEvent) => {
     if (!isMenuDisabled && e.button === 2) {
-      document.body.classList.add('no-selection');
+      (e.target as HTMLElement).classList.add('no-selection');
     }
   }, [isMenuDisabled]);
 
   const handleContextMenu = useCallback((e: React.MouseEvent) => {
-    document.body.classList.remove('no-selection');
+    (e.target as HTMLElement).classList.remove('no-selection');
 
     if (isMenuDisabled || (shouldDisableOnLink && (e.target as HTMLElement).matches('a[href]'))) {
       return;