fix highlight code