tests:
 - update for new API.