update to the test script