#31: Fix update UX