Fix for Swft cell class

DenTelezhkin · Aug 31, 2015 · ec0295a · ec0295a
1 parent ff97c1a
commit ec0295a
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