Fixed a gap in test coverage