Fix UI updates.