diff --git a/ios/lib/ReactNativeCameraKit/CKGalleryCollectionViewCell.m b/ios/lib/ReactNativeCameraKit/CKGalleryCollectionViewCell.m
index 6c9121b..790f4f1 100644
--- a/ios/lib/ReactNativeCameraKit/CKGalleryCollectionViewCell.m
+++ b/ios/lib/ReactNativeCameraKit/CKGalleryCollectionViewCell.m
@@ -294,11 +294,14 @@ static UIColor *selectionOverlayColor = nil;
         }
     }
     else {
+        self.imageOveray.backgroundColor = [UIColor clearColor];
         if (unSelectedImageIcon) {
-            self.imageOveray.backgroundColor = [UIColor clearColor];
             self.badgeImageView.image = unSelectedImageIcon;
             [self updateBadgeImageViewFrame];
         }
+        else {
+            self.badgeImageView.image = nil;
+        }
     }
 }