color fixups

1 files changed, 7 insertions, 6 deletions

diff --git a/pkg/commands/patch/patch_parser.go b/pkg/commands/patch/patch_parser.go
index 12426a3df..4fb8eb68b 100644
--- a/pkg/commands/patch/patch_parser.go
+++ b/pkg/commands/patch/patch_parser.go
@@ -118,15 +118,16 @@ func coloredString(textStyle style.TextStyle, str string, selected bool, include
 		textStyle = textStyle.MergeStyle(theme.SelectedRangeBgColor)
 	}
 
-	if len(str) < 2 {
-		return textStyle.Sprint(str)
+	firstCharStyle := textStyle
+	if included {
+		firstCharStyle = firstCharStyle.MergeStyle(style.BgGreen)
 	}
 
-	res := textStyle.Sprint(str[:1])
-	if included {
-		return res + textStyle.MergeStyle(style.BgGreen).Sprint(str[1:])
+	if len(str) < 2 {
+		return firstCharStyle.Sprint(str)
 	}
-	return res + textStyle.Sprint(str[1:])
+
+	return firstCharStyle.Sprint(str[:1]) + textStyle.Sprint(str[1:])
 }
 
 func parsePatch(patch string) ([]int, []int, []*PatchLine) {