Fix the button enable/disable logic again.