Modify audit test script