Fixed the test runner