Fixes test script.