ui tiny improvement