diff --git a/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/EditorCellFoldingBar.kt b/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/EditorCellFoldingBar.kt
index b8cfede9c8d2..6d4b4af2c820 100644
--- a/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/EditorCellFoldingBar.kt
+++ b/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/EditorCellFoldingBar.kt
@@ -47,8 +47,13 @@ class EditorCellFoldingBar(
     }
 
   fun dispose() {
-    panel?.let { editor.gutterComponentEx.remove(it) }
-    panel = null
+    panel?.let {
+      editor.gutterComponentEx.apply {
+        remove(it)
+        repaint()
+      }
+      panel = null
+    }
   }
 
   fun updateBounds() {