Silly me.
Fix highlighting