Adjust test script