From 29d87a88f08990e32b43d7be5a06f1cb9c410bd3 Mon Sep 17 00:00:00 2001
From: Rishabh Saxena <rishabh.robben@gmail.com>
Date: Mon, 14 Dec 2020 12:38:00 +0530
Subject: [PATCH] unmount during drag (#2168)

---
 .../src/components/editorComponents/DraggableComponent.tsx  | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/app/client/src/components/editorComponents/DraggableComponent.tsx b/app/client/src/components/editorComponents/DraggableComponent.tsx
index dfa0d53aaf..2f72f7b7c8 100644
--- a/app/client/src/components/editorComponents/DraggableComponent.tsx
+++ b/app/client/src/components/editorComponents/DraggableComponent.tsx
@@ -176,6 +176,10 @@ const DraggableComponent = (props: DraggableComponentProps) => {
 
   const className = `${classNameForTesting}`;
 
+  const shouldRenderComponent = !(
+    selectedWidget === props.widgetId && isDragging
+  );
+
   return (
     <DraggableWrapper
       className={className}
@@ -184,7 +188,7 @@ const DraggableComponent = (props: DraggableComponentProps) => {
       onClick={handleClick}
       style={style}
     >
-      {props.children}
+      {shouldRenderComponent && props.children}
       {widgetBoundaries}
     </DraggableWrapper>
   );