update to work with the newest test runner.