Minor highlight fix