Slight diff tweak