tutorial part 5 - fix tests