Fix highlight code