improving unit test coverage