Modified the test script