Fix test script