fixed highlighting bug