UElement を PsiElement として使用していることを報告します。
この方法で取得した PsiElement はあいまいです。
「物理的な」PsiElement を取得するには UElementKt.getSourcePsiElement() を使用し、Java 要素 (PsiClass、PsiMethod など) の動作を「エミュレートする」 PsiElement には
UElementKt.getAsJavaPsiElement() を使用してください。
詳しくは SDK ドキュメントの「UAST - Unified Abstract Syntax Tree」を参照してください。