fixing unit test