ui minor fix