Changing code to make the 100% test case coverage report