Fix button UI