Make AST module a little more elegant.