Fix the unit test.