Fix animation restart