Add more test coverage.