@Override
  public PsiElement adjustElementToCopy(@NotNull PsiElement element) {
    return element instanceof PyClass || element instanceof PyFunction ? element : null;
  }
@Override
  public PsiElement adjustElementToCopy(@NotNull PsiElement element)