fix tests for task 2