Overdue update for ui package API changes.