Fix highlight keywords in comment