fix exception on last level of stage (hotfix)