A little fix - Task 2