Let cell nib-files be found when this project is included in an app a…

…s a framework
jnutting · Aug 9, 2017 · a8fbf48 · a8fbf48
1 parent 8fd1b1b
commit a8fbf48
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/MultiProductViewer/TBTMultiProductViewController/TBTMultiProductViewController.m b/MultiProductViewer/TBTMultiProductViewController/TBTMultiProductViewController.m
@@ -59,9 +59,9 @@ - (void)viewDidLoad {
     [super viewDidLoad];
 
     // Register cells and supplementary views
-    [self.collectionView registerNib:[UINib nibWithNibName:@"TBTProductCell" bundle:nil]
+    [self.collectionView registerNib:[UINib nibWithNibName:@"TBTProductCell" bundle:[NSBundle bundleForClass:[self class]]]
           forCellWithReuseIdentifier:@"Product"];
-    [self.collectionView registerNib:[UINib nibWithNibName:@"TBTProductClusterHeaderCell" bundle:nil] forSupplementaryViewOfKind:UICollectionElementKindSectionHeader withReuseIdentifier:@"SectionHeader"];
+    [self.collectionView registerNib:[UINib nibWithNibName:@"TBTProductClusterHeaderCell" bundle:[NSBundle bundleForClass:[self class]]] forSupplementaryViewOfKind:UICollectionElementKindSectionHeader withReuseIdentifier:@"SectionHeader"];
 
     // Configure flow layout
     UICollectionViewFlowLayout *layout = (id)[self collectionViewLayout];