Fix a unit test bug.