Fix button display