Update for the interface change.