more unit test coverage