Fix AST usage