adapt tests to new API