fixed unit test issue