Fixed a bug with the animation