add better test coverage