Remove state associated with method argument in Rewriter