[iOS] Fix race in UI