Fix the test runner