diff --git a/idea/idea-jvm/src/org/jetbrains/kotlin/idea/scratch/actions/ScratchAction.kt b/idea/idea-jvm/src/org/jetbrains/kotlin/idea/scratch/actions/ScratchAction.kt index aff99969d1a..a1c64a19f79 100644 --- a/idea/idea-jvm/src/org/jetbrains/kotlin/idea/scratch/actions/ScratchAction.kt +++ b/idea/idea-jvm/src/org/jetbrains/kotlin/idea/scratch/actions/ScratchAction.kt @@ -15,7 +15,7 @@ import org.jetbrains.kotlin.idea.scratch.getScratchFileFromSelectedEditor import javax.swing.Icon abstract class ScratchAction(message: String, icon: Icon) : AnAction(message, message, icon) { - override fun update(e: AnActionEvent) = ApplicationManager.getApplication().invokeLater { + override fun update(e: AnActionEvent) { val scratchFile = e.getData(CommonDataKeys.EDITOR) ?.let { TextEditorProvider.getInstance().getTextEditor(it).getScratchFile() } ?: e.project?.let { getScratchFileFromSelectedEditor(it) }