fixed unit test faild