From a7249fea5a374ccdd3dd088430bd1d0740d5cc4f Mon Sep 17 00:00:00 2001
From: Anton Medvedev <anton@medv.io>
Date: Sun, 24 Sep 2023 13:21:18 +0200
Subject: [PATCH] Wrap collapsed nodes as well

---
 wrap.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/wrap.go b/wrap.go
index e55eee8..4b6ad8e 100644
--- a/wrap.go
+++ b/wrap.go
@@ -42,7 +42,11 @@ func wrapAll(n *node, termWidth int) {
 				}
 			}
 		}
-		n = n.next
+		if n.isCollapsed() {
+			n = n.collapsed
+		} else {
+			n = n.next
+		}
 	}
 }