Fix button loading states