Update UI fix logic