fix highlighting bug