updated test to use parse instead of raw AST