Merge pull request #11 from orkenstein/master

Fix for Swft cell class
DenTelezhkin · Sep 1, 2015 · 59a326f · 59a326f
2 parents ff97c1a + ec0295a
commit 59a326f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/DTCollectionViewManager/DTCollectionViewFactory.m b/DTCollectionViewManager/DTCollectionViewFactory.m
@@ -76,10 +76,12 @@ - (void)registerCellClass:(Class)cellClass forModelClass:(Class)modelClass
 - (void)registerNibNamed:(NSString *)nibName forCellClass:(Class)cellClass forModelClass:(Class)modelClass
 {
     NSParameterAssert([self nibExistsWithNibName:nibName]);
-
+
+    NSString * cellClassString = [DTRuntimeHelper classStringForClass:cellClass];
+
     [[self.delegate collectionView] registerNib:[UINib nibWithNibName:nibName bundle:[NSBundle mainBundle]]
                      forCellWithReuseIdentifier:[DTRuntimeHelper classStringForClass:cellClass]];
-    self.cellMappings[[DTRuntimeHelper modelStringForClass:modelClass]] = NSStringFromClass(cellClass);
+    self.cellMappings[[DTRuntimeHelper modelStringForClass:modelClass]] = cellClassString;
 }
 
 - (void)registerSupplementaryClass:(Class)supplementaryClass