fixing unit test interface