public static PsiElement returnToPhysical(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;
  }
public static PsiElement returnToPhysical(PsiElement element)