minor update to test script