fixing level 4 bug