Added extra test case to up code coverage