fix the unit test.