Improve test script.