image of the first selection is not appearing issue resolved

malekwasim · Sep 19, 2021 · a78c21c · a78c21c
1 parent 33c85a3
commit a78c21c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/WMSegmentControl/Source/WMSegment.swift b/WMSegmentControl/Source/WMSegment.swift
@@ -258,7 +258,7 @@ open class WMSegment: UIControl {
             }
 
             if btn == sender {
-                if let imageSelected = imagesSelected[safe: buttonIndex] {
+                if let imageSelected = imagesSelected[safe: buttonIndex], !imageSelected.isEmpty {
                     btn.setImage(UIImage(named: imageSelected), for: .normal)
                 }
                 selectedSegmentIndex = buttonIndex
@@ -291,7 +291,7 @@ open class WMSegment: UIControl {
             }
 
             if btn.tag == index {
-                if let imageSelected = imagesSelected[safe: buttonIndex] {
+                if let imageSelected = imagesSelected[safe: buttonIndex], !imageSelected.isEmpty {
                     btn.setImage(UIImage(named: imageSelected), for: .normal)
                 }
                 selectedSegmentIndex = buttonIndex