Modified the test code.