better unit test coverage