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 84a578d736..d68b73bf18 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
@@ -177,7 +177,9 @@ object UtTestsDialogProcessor {
                 }
         }
 
-        val filesToCompile = model.srcClasses.map { it.containingFile.virtualFile }.toTypedArray()
+        val filesToCompile = (model.srcClasses + listOfNotNull(springConfigClass))
+            .map { it.containingFile.virtualFile }
+            .toTypedArray()
 
         val promise = compile(project, filesToCompile, springConfigClass)
         promise.onSuccess {