From c7794a31dd0aa108578c24c2e4c468a12dd546b5 Mon Sep 17 00:00:00 2001
From: Adlai Holler <guy.on.internet@me.com>
Date: Mon, 3 Apr 2017 03:09:57 -0700
Subject: [PATCH] Beef up the headers for ASSectionController and
 ASSupplementaryNodeSource (#3233)

---
 Source/ASSectionController.h       | 4 ++++
 Source/ASSupplementaryNodeSource.h | 3 +--
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/Source/ASSectionController.h b/Source/ASSectionController.h
index dbf39d93c0..f57413f07d 100644
--- a/Source/ASSectionController.h
+++ b/Source/ASSectionController.h
@@ -7,9 +7,13 @@
 //
 
 #import <Foundation/Foundation.h>
+#import <AsyncDisplayKit/ASBlockTypes.h>
+#import <AsyncDisplayKit/ASDimension.h>
 
 NS_ASSUME_NONNULL_BEGIN
 
+@class ASBatchContext;
+
 /**
  * A protocol that your section controllers should conform to,
  * in addition to IGListSectionType, in order to be used with AsyncDisplayKit.
diff --git a/Source/ASSupplementaryNodeSource.h b/Source/ASSupplementaryNodeSource.h
index 9d3233c9e2..52ad429377 100644
--- a/Source/ASSupplementaryNodeSource.h
+++ b/Source/ASSupplementaryNodeSource.h
@@ -7,12 +7,11 @@
 //
 
 #import <Foundation/Foundation.h>
+#import <AsyncDisplayKit/ASBlockTypes.h>
 #import <AsyncDisplayKit/ASDimension.h>
 
 NS_ASSUME_NONNULL_BEGIN
 
-@class ASCellNode;
-
 @protocol ASSupplementaryNodeSource <NSObject>
 
 /**