Update test to new API.