Animation restart improvement