fixed unit test partially