fix bug in task 9