fixed unit test
fixed unit test