Update to latest ui API.