now fix the animation properly