From 7c13dee33d52bedcf2518b63b779276f9f251f02 Mon Sep 17 00:00:00 2001
From: Zachary Yedidia <zyedidia@gmail.com>
Date: Thu, 11 Aug 2016 16:52:50 -0400
Subject: [PATCH] Small fix to splits

---
 cmd/micro/split_tree.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/cmd/micro/split_tree.go b/cmd/micro/split_tree.go
index d1c5bc33..fed4490c 100644
--- a/cmd/micro/split_tree.go
+++ b/cmd/micro/split_tree.go
@@ -61,6 +61,7 @@ func (l *LeafNode) VSplit(buf *Buffer) {
 		newView.Num = len(tab.views)
 		s.children = []Node{l, NewLeafNode(newView, s)}
 		l.parent.children[search(l.parent.children, l)] = s
+		l.parent = s
 
 		tab.curView++
 		tab.views = append(tab.views, newView)
@@ -86,6 +87,7 @@ func (l *LeafNode) HSplit(buf *Buffer) {
 		newView.Num = len(tab.views)
 		s.children = []Node{l, NewLeafNode(newView, s)}
 		l.parent.children[search(l.parent.children, l)] = s
+		l.parent = s
 
 		tab.curView++
 		tab.views = append(tab.views, newView)
@@ -119,6 +121,7 @@ func (s *SplitTree) Cleanup() {
 			if len(n.children) == 1 {
 				if _, ok := n.children[0].(*LeafNode); ok {
 					s.children[i] = n.children[0]
+					continue
 				}
 			}
 			n.Cleanup()