public @Nullable PsiNamedElement getProblemElement(@NotNull PsiElement psiElement) {
    return psiElement.getContainingFile();
  }
public @Nullable PsiNamedElement getProblemElement(@NotNull PsiElement psiElement)