Fix update UI bug