diff --git a/src/ui/nehubaContainer/nehubaContainer.component.ts b/src/ui/nehubaContainer/nehubaContainer.component.ts
index 2b194f08441a451af6fde30a42fa0578becfec09..5aad100330017ba1a7902110afa48c5aac00258d 100644
--- a/src/ui/nehubaContainer/nehubaContainer.component.ts
+++ b/src/ui/nehubaContainer/nehubaContainer.component.ts
@@ -776,15 +776,6 @@ export class NehubaContainer implements OnInit, OnChanges, OnDestroy {
         }
       }),
     )
-
-    // To get WebGL content when taking screenshot
-    HTMLCanvasElement.prototype.getContext = function(origFn) {
-      return function(type, attribs) {
-        attribs = attribs || {}
-        attribs.preserveDrawingBuffer = true
-        return origFn.call(this, type, attribs)
-      }
-    }(HTMLCanvasElement.prototype.getContext)
   }
 
   // datasetViewerRegistry : Set<string> = new Set()
diff --git a/src/ui/takeScreenshot/takeScreenshot.component.ts b/src/ui/takeScreenshot/takeScreenshot.component.ts
index ca8ce997233a9715eea067c2786587cf96368ed6..b10f94a06cee887ece6d331b661ec137af58a15d 100644
--- a/src/ui/takeScreenshot/takeScreenshot.component.ts
+++ b/src/ui/takeScreenshot/takeScreenshot.component.ts
@@ -189,6 +189,7 @@ export class TakeScreenshotComponent implements OnInit, OnDestroy {
     }
 
     public mouseUp(_event: MouseEvent) {
+      (window as any).viewer.display.update()
       this.borderWidth = '0'
 
       this.isDragging = false