TODO fix toggle switch