Fix minor bug in test runner