Fix test coverage.