better ui fix

ui fix