Fixed problem with animation