From a45aa360dda51b1156a88fd1655bb72d0aa27470 Mon Sep 17 00:00:00 2001
From: Ayman Bagabas <ayman.bagabas@gmail.com>
Date: Tue, 21 Mar 2023 15:02:15 -0400
Subject: [PATCH] Update logger.go

Co-authored-by: Carlos Alexandro Becker <caarlos0@users.noreply.github.com>
---
 logger.go | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/logger.go b/logger.go
index 2696527..2794870 100644
--- a/logger.go
+++ b/logger.go
@@ -255,11 +255,13 @@ func (l *Logger) SetOutput(w io.Writer) {
 	atomic.StoreUint32(&l.isDiscard, isDiscard)
 	// Reuse cached renderers
 	v, ok := registry.Load(w)
-	if !ok {
+	if ok {
+		l.re = v.(*lipgloss.Renderer)
+	} else {
 		v = lipgloss.NewRenderer(w, termenv.WithColorCache(true))
 		registry.Store(w, v)
+		l.re = v
 	}
-	l.re = v.(*lipgloss.Renderer)
 }
 
 // SetFormatter sets the formatter.