Fix UI after unit model changes