diff --git a/src/UIViewController+TransitionController.h b/src/UIViewController+TransitionController.h
index 1feaf84..b1f5279 100644
--- a/src/UIViewController+TransitionController.h
+++ b/src/UIViewController+TransitionController.h
@@ -15,6 +15,7 @@
  */
 
 #import <Foundation/Foundation.h>
+#import <UIKit/UIKit.h>
 
 @protocol MDMTransitionController;
 
diff --git a/src/private/MDMPresentationTransitionController.h b/src/private/MDMPresentationTransitionController.h
index b620bba..22fe219 100644
--- a/src/private/MDMPresentationTransitionController.h
+++ b/src/private/MDMPresentationTransitionController.h
@@ -15,6 +15,7 @@
  */
 
 #import <Foundation/Foundation.h>
+#import <UIKit/UIKit.h>
 
 #import "MDMTransitionController.h"