adapt to api changes in test framework