Small fix for test runner script