Fix a stage bug