Improve test coverage.