[IMP]improve code remove method from onchange_stage_id