Increase test case coverage