Fix test script.