Update UI and register bug