Modified the test case