diff --git a/notebooks/visualization/src/com/intellij/notebooks/visualization/NotebookVisualizationUiUtil.kt b/notebooks/visualization/src/com/intellij/notebooks/visualization/NotebookVisualizationUiUtil.kt
index 93583590b8c4..30b63a2a543d 100644
--- a/notebooks/visualization/src/com/intellij/notebooks/visualization/NotebookVisualizationUiUtil.kt
+++ b/notebooks/visualization/src/com/intellij/notebooks/visualization/NotebookVisualizationUiUtil.kt
@@ -24,7 +24,7 @@ inline fun Graphics.useG2D(handler: (Graphics2D) -> Unit) {
     handler(g2d)
   }
   finally {
-    dispose()
+    g2d.dispose()
   }
 }