Fix buttons enabled/disabled