fix - unit test fail