- fixed unit test