Note all details that led up to discipline

Having good documentation of your reasoning will often persuade a judge or jury that discrimination wasn’t the real reason for differing discipline, and that you legitimately used discretion to arrive at the appropriate punishment.