Fix for levelstage