Fixing the button disable