Adding test code trying to improve test coverage.