Updated the test script.