Use AstMapVisitor instead of ExprMapVisitor as needed