Fix UKMS toggle