Fix broken --ast option