fixing unit test errors