Fix button animation