small modification to test script