Fix level increase code