Fix bug in unit test for problem 1.