diff --git a/src/main/java/com/zhongan/devpilot/util/PsiElementUtils.java b/src/main/java/com/zhongan/devpilot/util/PsiElementUtils.java
index a7801cf..c023976 100644
--- a/src/main/java/com/zhongan/devpilot/util/PsiElementUtils.java
+++ b/src/main/java/com/zhongan/devpilot/util/PsiElementUtils.java
@@ -59,7 +59,7 @@ private static Set<PsiClass> getClassRelatedClass(@NotNull PsiElement element) {
             var fields = psiClass.getFields();
 
             for (PsiMethod psiMethod : methods) {
-                result.addAll(getMethodParameterTypeClass(psiMethod));
+                result.addAll(getMethodRelatedClass(psiMethod));
             }
 
             for (PsiField psiField : fields) {