Fixed small bug with animation