diff --git a/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/DecoratedEditor.kt b/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/DecoratedEditor.kt
index 61780b450d35..7c1bd0b5a6e8 100644
--- a/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/DecoratedEditor.kt
+++ b/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/DecoratedEditor.kt
@@ -1,6 +1,7 @@
 package org.jetbrains.plugins.notebooks.visualization.ui
 
 import com.intellij.ide.structureView.StructureViewBuilder
+import com.intellij.openapi.application.ApplicationManager
 import com.intellij.openapi.application.ReadAction
 import com.intellij.openapi.client.ClientSystemInfo
 import com.intellij.openapi.editor.Caret
@@ -95,7 +96,7 @@ private class DecoratedEditor(private val original: TextEditor, private val mana
 
   private fun scheduleSelectionUpdate() {
     if (selectionUpdateScheduled.compareAndSet(false, true)) {
-      SwingUtilities.invokeLater {
+      ApplicationManager.getApplication().invokeLater {
         try {
           updateSelectionByCarets()
         }
diff --git a/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/NotebookEditorUiUtil.kt b/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/NotebookEditorUiUtil.kt
index 90b2f0b0aae1..40a34ecc61f9 100644
--- a/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/NotebookEditorUiUtil.kt
+++ b/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/NotebookEditorUiUtil.kt
@@ -1,6 +1,7 @@
 package org.jetbrains.plugins.notebooks.visualization.ui
 
 import com.intellij.openapi.application.ReadAction
+import com.intellij.openapi.application.WriteIntentReadAction
 import com.intellij.openapi.editor.Editor
 import com.intellij.openapi.editor.EditorKind
 import com.intellij.openapi.editor.Inlay
@@ -57,7 +58,7 @@ fun EditorEx.addComponentInlay(
 private fun updateUiOnParentResizeImpl(parent: JComponent, childRef: WeakReference<JComponent>) {
   val listener = object : ComponentAdapter() {
     override fun componentResized(e: ComponentEvent?) {
-      ReadAction.run<Throwable> {
+      WriteIntentReadAction.run {
         val child = childRef.get()
         if (child != null) {
           child.updateUI()