Tweak test runner