Fixing the code highlights