Fix test runner error