Fix test runner errors