fix exception on last level of stage