fix : minor ui fix