UI minor fix