Fix line highlighting