update for new changes to AST