Use _ASCollectionReusableView inside ASIGListSupplementaryViewSourceM…

…ethods (#40)
TextureGroup · Apr 19, 2017 · dca67c3 · dca67c3
1 parent ed5a139
commit dca67c3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/Source/AsyncDisplayKit+IGListKitMethods.m b/Source/AsyncDisplayKit+IGListKitMethods.m
@@ -13,6 +13,8 @@
 #import "AsyncDisplayKit+IGListKitMethods.h"
 #import <AsyncDisplayKit/ASAssert.h>
 #import <AsyncDisplayKit/_ASCollectionViewCell.h>
+#import <AsyncDisplayKit/_ASCollectionReusableView.h>
+
 
 @implementation ASIGListSectionControllerMethods
 
@@ -35,7 +37,7 @@ + (__kindof UICollectionReusableView *)viewForSupplementaryElementOfKind:(NSStri
                                                                  atIndex:(NSInteger)index
                                                        sectionController:(IGListSectionController<IGListSectionType> *)sectionController
 {
-  return [sectionController.collectionContext dequeueReusableSupplementaryViewOfKind:elementKind forSectionController:sectionController class:[UICollectionReusableView class] atIndex:index];
+  return [sectionController.collectionContext dequeueReusableSupplementaryViewOfKind:elementKind forSectionController:sectionController class:[_ASCollectionReusableView class] atIndex:index];
 }
 
 + (CGSize)sizeForSupplementaryViewOfKind:(NSString *)elementKind atIndex:(NSInteger)index