Bug with line highlighting fixed