PsiElement (PsiElement element) { if (element instanceof LightElement) { UElement uElement = UastContextKt.toUElement(element); PsiElement el = uElement == null ? null : uElement.getSourcePsi(); if (el != null) return el; return element.getNavigationElement(); } return element; }