improve unit test coverage