Update ui on upgrade