Fix button layout