Fix animation on mobile