from last - fix to remove overlay elements (doucment icons)

1 files changed, 6 insertions, 6 deletions

diff --git a/src/client/views/OverlayView.tsx b/src/client/views/OverlayView.tsx
index e41113ca4..673432a60 100644
--- a/src/client/views/OverlayView.tsx
+++ b/src/client/views/OverlayView.tsx
@@ -143,11 +143,7 @@ export class OverlayView extends React.Component {
 
     @action
     addElement(ele: JSX.Element, options: OverlayElementOptions): OverlayDisposer {
-        const remove = action(() => {
-            const index = this._elements.indexOf(ele);
-            if (index !== -1) this._elements.splice(index, 1);
-        });
-        this._elements.push(
+        const div = (
             <div
                 key={Utils.GenerateGuid()}
                 className="overlayView-wrapperDiv"
@@ -161,7 +157,11 @@ export class OverlayView extends React.Component {
                 {ele}
             </div>
         );
-        return remove;
+        this._elements.push(div);
+        return action(() => {
+            const index = this._elements.indexOf(div);
+            if (index !== -1) this._elements.splice(index, 1);
+        });
     }
 
     @action