diff --git a/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/NotebookCellInlayManager.kt b/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/NotebookCellInlayManager.kt
index c2b86b884eb6..161989343bc2 100644
--- a/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/NotebookCellInlayManager.kt
+++ b/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/NotebookCellInlayManager.kt
@@ -91,6 +91,9 @@ class NotebookCellInlayManager private constructor(val editor: EditorImpl) : Not
             // Many UI instances has overridden getPreferredSize relying on editor dimensions.
             inlay.renderer?.asSafely<JComponent>()?.updateUI()
           }
+          _cells.forEach {
+            it.onViewportChange()
+          }
         }
       })
     }
diff --git a/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/EditorCell.kt b/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/EditorCell.kt
index eb3be7fc07b7..5a4a2c5725af 100644
--- a/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/EditorCell.kt
+++ b/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/EditorCell.kt
@@ -114,6 +114,11 @@ class EditorCell(
     output?.updatePositions()
   }
 
+  fun onViewportChange() {
+    input.onViewportChange()
+    output?.onViewportChange()
+  }
+
   companion object {
     private val LOG = logger<EditorCell>()
   }
diff --git a/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/EditorCellInput.kt b/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/EditorCellInput.kt
index 1cd317597fab..ec5e98264009 100644
--- a/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/EditorCellInput.kt
+++ b/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/EditorCellInput.kt
@@ -83,6 +83,10 @@ internal class EditorCellInput(
   fun updatePositions() {
     folding.updatePosition()
   }
+
+  fun onViewportChange() {
+    inputController?.onViewportChange()
+  }
 }
 
 private fun String.ellipsis(length: Int): String {
diff --git a/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/EditorCellOutput.kt b/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/EditorCellOutput.kt
index 75ee8eaa587b..acbe53b8d1d7 100644
--- a/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/EditorCellOutput.kt
+++ b/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/EditorCellOutput.kt
@@ -17,4 +17,8 @@ internal class EditorCellOutput(editor: EditorEx, private val outputController:
     folding.dispose()
   }
 
+  fun onViewportChange() {
+    outputController.onViewportChange()
+  }
+
 }
\ No newline at end of file