Fixup to test script