small bug fix to level 4