fix stage bug