Minor update to test script