Expanding test coverage to 100% for existing tests