diff --git a/plugins/kotlin/idea/src/org/jetbrains/kotlin/idea/script/k2/utils.kt b/plugins/kotlin/idea/src/org/jetbrains/kotlin/idea/script/k2/utils.kt
index 3b37d0be03a2..60e6c5efed34 100644
--- a/plugins/kotlin/idea/src/org/jetbrains/kotlin/idea/script/k2/utils.kt
+++ b/plugins/kotlin/idea/src/org/jetbrains/kotlin/idea/script/k2/utils.kt
@@ -9,7 +9,9 @@ import com.intellij.psi.SyntaxTraverser
 import com.intellij.util.containers.TreeTraversal
 import org.jetbrains.kotlin.lexer.KtTokens.SHEBANG_COMMENT
 
-fun isMainKtsScript(virtualFile: VirtualFile) = virtualFile.name.endsWith(".main.kts")
+private const val MAIN_KTS = "main.kts"
+
+fun isMainKtsScript(virtualFile: VirtualFile) = virtualFile.name == MAIN_KTS || virtualFile.name.endsWith(".$MAIN_KTS")
 
 fun PsiFile.hasShebangComment(): Boolean =
     SyntaxTraverser.psiTraverser(/* root = */ this)