Fix test script code.