Fixed YHButton logic error