Fix comment highlight

ref #11