Increasing test coverage slightly