Fixed issue that was preventing 100% code coverage