fix tests for week5