Small change on the button disable end