Bugfix in stage