fix change stage bug