Fixing test script.