Update ui_utils.go