fixed tests for the first task