Fixed button bug

oops