UI update fix