fixe tests for task 4