Fix full range value (#2)

ElonPark · May 4, 2021 · 6f231be · 6f231be
1 parent 9d77afc
commit 6f231be
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/AttributedStringSugar/Classes/AttributedStringSugar.swift b/AttributedStringSugar/Classes/AttributedStringSugar.swift
@@ -61,7 +61,7 @@ public extension NSMutableAttributedString {
         if let range = range {
             self.addAttribute(key, value: value, range: range)
         } else {
-            let range = NSRange(location: 0, length: self.string.count)
+            let range = NSRange(location: 0, length: length)
             self.addAttribute(key, value: value, range: range)
         }
 
@@ -86,7 +86,7 @@ public extension NSMutableAttributedString {
         if let range = range {
             self.addAttributes(attrs, range: range)
         } else {
-            let range = NSRange(location: 0, length: self.string.count)
+            let range = NSRange(location: 0, length: length)
             self.addAttributes(attrs, range: range)
         }