fixed unit test fail