diff --git a/kotlin/src/main/kotlin/cc/unitmesh/kotlin/provider/KotlinWriteTestService.kt b/kotlin/src/main/kotlin/cc/unitmesh/kotlin/provider/KotlinWriteTestService.kt
index 6b0f945180..f9fa7d822a 100644
--- a/kotlin/src/main/kotlin/cc/unitmesh/kotlin/provider/KotlinWriteTestService.kt
+++ b/kotlin/src/main/kotlin/cc/unitmesh/kotlin/provider/KotlinWriteTestService.kt
@@ -38,7 +38,7 @@ class KotlinWriteTestService : WriteTestService() {
         val className = sourceFile.name.replace(".kt", "") + "Test"
 
         val packageName = ReadAction.compute<String, Throwable> {
-            (sourceFile as PsiJavaFile).packageName
+            (sourceFile as KtFile).packageName
         }
         val parentDirPath = ReadAction.compute<String, Throwable> {
             parentDir?.path