Fix code hightlight