fix comment highlighting