Fix error in test runner