fix the unit test failure