Small fix to test runner.