ui: small fix