AST: Fix usage of is_lax