diff --git a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/UtTestsDialogProcessor.kt b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/UtTestsDialogProcessor.kt
index 78a8f6209b..843a443055 100644
--- a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/UtTestsDialogProcessor.kt
+++ b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/UtTestsDialogProcessor.kt
@@ -162,7 +162,7 @@ object UtTestsDialogProcessor {
                             )
 
                             for (srcClass in model.srcClasses) {
-                                val methods = ReadAction.nonBlocking<List<ExecutableId>> {
+                                val (methods, className) = ReadAction.nonBlocking<Pair<List<ExecutableId>, String?>> {
                                     val canonicalName = srcClass.canonicalName
                                     val clazz = classLoader.loadClass(canonicalName).kotlin
                                     psi2KClass[srcClass] = clazz
@@ -180,10 +180,9 @@ object UtTestsDialogProcessor {
                                         clazz.allNestedClasses.flatMap {
                                             findMethodsInClassMatchingSelected(it, srcMethods)
                                         }
-                                    })
+                                    }) to srcClass.name
                                 }.executeSynchronously()
 
-                                val className = srcClass.name
                                 if (methods.isEmpty()) {
                                     logger.error { "No methods matching selected found in class $className." }
                                     continue