Task 4 quick fix