Fixed code snippet highlighting