Modify test script