diff --git a/idea/idea-jvm/src/org/jetbrains/kotlin/idea/injection/KotlinLanguageInjectionSupport.kt b/idea/idea-jvm/src/org/jetbrains/kotlin/idea/injection/KotlinLanguageInjectionSupport.kt index c429733bb4a..503a843c8e5 100644 --- a/idea/idea-jvm/src/org/jetbrains/kotlin/idea/injection/KotlinLanguageInjectionSupport.kt +++ b/idea/idea-jvm/src/org/jetbrains/kotlin/idea/injection/KotlinLanguageInjectionSupport.kt @@ -76,15 +76,16 @@ class KotlinLanguageInjectionSupport : AbstractLanguageInjectionSupport() { val project = psiElement.getProject() - val injectionAnnotation = findAnnotationInjection(psiElement) - val injectionComment = findInjectionComment(psiElement) - - val injectInstructions = listOf(injectionAnnotation, injectionComment).filterNotNull() - - val configuration = Configuration.getProjectInstance(project) + val injectInstructions = listOfNotNull( + findAnnotationInjection(psiElement), + findInjectionComment(psiElement) + ) TemporaryPlacesRegistry.getInstance(project).removeHostWithUndo(project, psiElement) - configuration.replaceInjectionsWithUndo(project, listOf(), listOf(), injectInstructions) + + project.executeWriteCommand("remove injection in-code instructions") { + injectInstructions.forEach(PsiElement::delete) + } return true }