Fixed issue with animation