Yet another test script fix