Fixed the task on Test 2