small modification to ast