minor ui bug