fixed unit test error