Fix snippet highlighting