Update logger.go

Co-authored-by: Carlos Alexandro Becker <[email protected]>
charmbracelet · Mar 21, 2023 · 023bb73 · 023bb73
1 parent 01f932d
commit 023bb73
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/logger.go b/logger.go
@@ -254,12 +254,12 @@ func (l *Logger) SetOutput(w io.Writer) {
 	}
 	atomic.StoreUint32(&l.isDiscard, isDiscard)
 	// Reuse cached renderers
-	v, ok := registry.Load(w)
-	if !ok {
-		v = lipgloss.NewRenderer(w, termenv.WithColorCache(true))
-		registry.Store(w, v)
+	if v, ok := registry.Load(w); ok {
+		l.re = v.(*lipgloss.Renderer)
+	} else {
+		l.re = lipgloss.NewRenderer(w, termenv.WithColorCache(true))
+		registry.Store(w, l.re)
 	}
-	l.re = v.(*lipgloss.Renderer)
 }
 
 // SetFormatter sets the formatter.