Task 2 Fix