fix ast transformation