fixed unit test

pass