Task 10 fix