Fix test script error