From a0ff04791150d5f4e52614eeac20c2654b62fcba Mon Sep 17 00:00:00 2001
From: Zachary Yedidia <zyedidia@gmail.com>
Date: Sun, 19 Feb 2017 11:35:23 -0500
Subject: [PATCH] Minor fixes

---
 highlighter.go | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/highlighter.go b/highlighter.go
index 34e86c4..a60aa19 100644
--- a/highlighter.go
+++ b/highlighter.go
@@ -216,6 +216,11 @@ func (h *Highlighter) ReHighlightLine(input LineStates, lineN int) {
 
 	line := []byte(lines[lineN])
 
+	h.lastRegion = nil
+	if lineN > 0 {
+		h.lastRegion = input.State(lineN - 1)
+	}
+
 	var match LineMatch
 	if lineN == 0 || h.lastRegion == nil {
 		match = h.highlightEmptyRegion(0, true, lineN, line)
@@ -247,11 +252,11 @@ func (h *Highlighter) ReHighlight(input LineStates, startline int) {
 		curState := h.lastRegion
 		lastState := input.State(i)
 
+		input.SetMatch(i, match)
+		input.SetState(i, curState)
+
 		if curState == lastState {
 			break
 		}
-
-		input.SetMatch(i, match)
-		input.SetState(i, curState)
 	}
 }