fix minor stage bug