diff --git a/kotlin/src/main/kotlin/cc/unitmesh/kotlin/context/KotlinClassContextBuilder.kt b/kotlin/src/main/kotlin/cc/unitmesh/kotlin/context/KotlinClassContextBuilder.kt
index 9966f537dc..c162a2ffe3 100644
--- a/kotlin/src/main/kotlin/cc/unitmesh/kotlin/context/KotlinClassContextBuilder.kt
+++ b/kotlin/src/main/kotlin/cc/unitmesh/kotlin/context/KotlinClassContextBuilder.kt
@@ -28,7 +28,8 @@ class KotlinClassContextBuilder : ClassContextBuilder {
         val usages =
             if (gatherUsages) JavaContextCollection.findUsages(psiElement as PsiNameIdentifierOwner) else emptyList()
 
-        return ClassContext(psiElement, text, name, ktNamedFunctions, allFields, null, usages)
+        val displayName = psiElement.fqName?.asString() ?: psiElement.name ?: ""
+        return ClassContext(psiElement, text, name, ktNamedFunctions, allFields, null, usages, displayName = displayName)
     }
 
     companion object {
diff --git a/kotlin/src/test/kotlin/cc/unitmesh/kotlin/provider/KotlinTestDataBuilderTest.kt b/kotlin/src/test/kotlin/cc/unitmesh/kotlin/provider/KotlinTestDataBuilderTest.kt
index ae43a18768..cf72005fcb 100644
--- a/kotlin/src/test/kotlin/cc/unitmesh/kotlin/provider/KotlinTestDataBuilderTest.kt
+++ b/kotlin/src/test/kotlin/cc/unitmesh/kotlin/provider/KotlinTestDataBuilderTest.kt
@@ -44,7 +44,7 @@ class KotlinTestDataBuilderTest : LightPlatformTestCase() {
 
         assertEquals(outboundData.size, 1)
         assertEquals(
-            outboundData["cc.unitmesh.untitled.demo.controller.UserDTO"], "'package: null\n" +
+            outboundData["cc.unitmesh.untitled.demo.controller.UserDTO"], "'package: cc.unitmesh.untitled.demo.controller.UserDTO\n" +
                     "class UserDTO {\n" +
                     "  \n" +
                     "  \n" +