challenge 04: fix tests