Small highlight fix