Small animation fix