diff --git a/libs/ui/CountersSelector.lua b/libs/ui/CountersSelector.lua
index be43e60..87ab1e1 100644
--- a/libs/ui/CountersSelector.lua
+++ b/libs/ui/CountersSelector.lua
@@ -6,20 +6,21 @@ end
 
 local function createWriteWithColor(textColor, backgroundColor, givenWin)
   local win = givenWin or term
-  local originalTextColor = nil
-  local originalBackgroundColor = nil
-
-  if textColor then
-    originalTextColor = win.getTextColor()
-    win.setTextColor(textColor)
-  end
-
-  if backgroundColor then
-    originalBackgroundColor = win.getBackgroundColor()
-    win.setBackgroundColor(backgroundColor)
-  end
 
   return function(str)
+    local originalTextColor = nil
+    local originalBackgroundColor = nil
+
+    if textColor then
+      originalTextColor = win.getTextColor()
+      win.setTextColor(textColor)
+    end
+
+    if backgroundColor then
+      originalBackgroundColor = win.getBackgroundColor()
+      win.setBackgroundColor(backgroundColor)
+    end
+
     win.write(str)
 
     if originalTextColor then