Readd AST rewrite rule