Fix test runner