  Enhanced Gaussian noise models are used to demonstrate that the per-block SNR
after fibre transmission varies significantly due to the variable-composition
nature of modern probabilistic shaping schemes. We propose a
nonlinearity-optimised distribution matcher that improves the average and
worst-case SNR by 0.14 and 0.22 dB, respectively.
