Fixed up bug with button anim