Fix animation bug