fix bug: highlight code