Fixed problem with animations (I think)