Fix small bug with selected lines highlighting