better test coverage