fixing unit test api