Small fix for animation