Small ui fix

Signed-off-by: Dmitriy Zaporozhets <dmitriy.zaporozhets@gmail.com>