RETURN AST transform works