task5 bug fix