Added test case to improve code coverage