Update test runner to use the new DiscoverRunner