Tweak test script