fixed stage bug