minor ui fix