public static @Nullable PsiElement dereference(PsiElement element) {
    if (element instanceof PyReferenceExpression) {
      return element.getReference().resolve();
    }
    return element;
  }
public static @Nullable PsiElement dereference(PsiElement element)