fix ast transformation
Merge pull request #1 from jeffbrown/ast_fix

fix ast transformation