fixed unit test.