load icon images from correct bundle

terryworona · Jan 24, 2017 · ea679b4 · ea679b4
1 parent 16f8e5c
commit ea679b4
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/Classes/TWMessageBarManager.m b/Classes/TWMessageBarManager.m
@@ -807,16 +807,23 @@ - (nonnull UIColor *)strokeColorForMessageType:(TWMessageBarMessageType)type
 - (nonnull UIImage *)iconImageForMessageType:(TWMessageBarMessageType)type
 {
     UIImage *iconImage = nil;
+    NSBundle *bundle = [NSBundle bundleForClass:[self class]];
     switch (type)
     {
         case TWMessageBarMessageTypeError:
-            iconImage = [UIImage imageNamed:kTWMessageBarStyleSheetImageIconError];
+            iconImage = [UIImage imageNamed:kTWMessageBarStyleSheetImageIconError
+                                   inBundle:bundle
+              compatibleWithTraitCollection:nil];
             break;
         case TWMessageBarMessageTypeSuccess:
-            iconImage = [UIImage imageNamed:kTWMessageBarStyleSheetImageIconSuccess];
+            iconImage = [UIImage imageNamed:kTWMessageBarStyleSheetImageIconSuccess
+                                   inBundle:bundle
+              compatibleWithTraitCollection:nil];
             break;
         case TWMessageBarMessageTypeInfo:
-            iconImage = [UIImage imageNamed:kTWMessageBarStyleSheetImageIconInfo];
+            iconImage = [UIImage imageNamed:kTWMessageBarStyleSheetImageIconInfo
+                                   inBundle:bundle
+              compatibleWithTraitCollection:nil];
             break;
     }
     return iconImage;