Unit test runner tweak