ui: fix a few UI update regressions