small changes to test script