fixed my unit test