diff --git a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/ui/actions/GenerateTestsAction.kt b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/ui/actions/GenerateTestsAction.kt
index 635b347ec6..cd14a1c363 100644
--- a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/ui/actions/GenerateTestsAction.kt
+++ b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/ui/actions/GenerateTestsAction.kt
@@ -61,7 +61,7 @@ class GenerateTestsAction : AnAction(), UpdateInBackground {
                 val module = ModuleUtil.findModuleForFile(srcSourceRoot, project) ?: return null
                 val matchingRoot = ModuleRootManager.getInstance(module).contentEntries
                     .flatMap { entry -> entry.sourceFolders.toList() }
-                    .singleOrNull { folder -> folder.file == srcSourceRoot }
+                    .firstOrNull { folder -> folder.file == srcSourceRoot }
                 if (srcMembers.isEmpty() || matchingRoot == null || matchingRoot.rootType.isForTests) {
                     return null
                 }