grammar: forgot to update the AST treewalker too