fix the test script