Fix: update UI after changes