minor fix to test script