Fix issue with animated button