Fix the mate weighting