Adding to unit test coverage