[IMP] change stages mechanisam