Refactor to slightly improve test coverage