fixed unit test failure