@Override
  public boolean isValidInContext(PsiElement element) {
    if (element instanceof PsiClass psiClass) {
      return InheritanceUtil.isInheritor(psiClass, BASE_CLASS);
    }
    return false;
  }
@Override
  public boolean isValidInContext(PsiElement element)