task 03: fix tests