Fixed AST processing to reflect changes in the AST structure after grammar overhaul.