Fix animation button reset after animation close/open