Modified the test case for better code coverage