Fix the test script