Bugfix: stage not recognized