@Override
  public void applyFix(@NotNull Project project, @NotNull PsiElement element, @NotNull ModPsiUpdater updater) {
    assert element instanceof PyCallExpression;
    if (element instanceof PyDecorator) {
      element.delete();
    }
    else {
      final PyArgumentList argumentList = ((PyCallExpression)element).getArgumentList();
      assert argumentList != null;
      argumentList.delete();
      final PyElementGenerator elementGenerator = PyElementGenerator.getInstance(project);
      // regenerate element for Psi consistency, because it isn't PyCallExpression anymore
      final PyExpression expression = elementGenerator.createExpressionFromText(LanguageLevel.forElement(element), element.getText());
      element.replace(expression);
    }
  }
@Override
  public void applyFix(@NotNull Project project, @NotNull PsiElement element, @NotNull ModPsiUpdater updater)