adjusted tests to work with the slightly altered AST format