Task 4 fix