fix task 2 bug