fixes highlighting bug