Correct diffraction run axis unit check

Refs #10094