diff --git a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/IntentionHelper.kt b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/IntentionHelper.kt
index 59ca4266eb..6cb224bd97 100644
--- a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/IntentionHelper.kt
+++ b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/IntentionHelper.kt
@@ -24,10 +24,10 @@ class IntentionHelper(val project: Project, private val editor: Editor, private
         val actions =
             DumbService.getInstance(project).runReadActionInSmartMode(Computable<Map<IntentionAction, String>> {
                 val daemonProgressIndicator = DaemonProgressIndicator()
-                Disposer.register(project, daemonProgressIndicator)//check it
-                val list = ProgressManager.getInstance().runProcess(Computable<List<HighlightInfo>> {
+                Disposer.register(project) { daemonProgressIndicator.cancel() }//check it
+                val list = ProgressManager.getInstance().runProcess(Computable<List<HighlightInfo>> inner@{
                     try {
-                        val containingFile = testFile.containingFile ?: return@Computable emptyList()
+                        val containingFile = testFile.containingFile ?: return@inner emptyList()
                         DaemonCodeAnalyzerEx.getInstanceEx(project).runMainPasses(
                             containingFile,
                             editor.document,