Fixed an ui bug