fix unit test failure