fixed ast bug