Fix code hilight