diff --git a/Makefile b/Makefile
index 85e96e2..ff84095 100644
--- a/Makefile
+++ b/Makefile
@@ -9,7 +9,7 @@ CONFIG_BRANCH = $(shell git branch --show-current)
 CONFIG_COMMIT = $(shell git log --oneline | sed '2,10000000d' | cut -b 1-7)
 
 # Build the UI library
-LiveContainerUI_FILES = LCAppDelegate.m LCJITLessSetupViewController.m LCRootViewController.m LCSettingsListController.m LCTabBarController.m LCUtils.m MBRoundProgressView.m unarchive.m AppInfo.m
+LiveContainerUI_FILES = LCAppDelegate.m LCJITLessSetupViewController.m LCRootViewController.m LCSettingsListController.m LCTabBarController.m LCUtils.m MBRoundProgressView.m UIViewController+LCAlert.m unarchive.m AppInfo.m
 LiveContainerUI_CFLAGS = \
   -fobjc-arc \
   -DCONFIG_TYPE=\"$(CONFIG_TYPE)\" \
diff --git a/UIViewController+LCAlert.m b/UIViewController+LCAlert.m
index 30aa9db..2088dd0 100644
--- a/UIViewController+LCAlert.m
+++ b/UIViewController+LCAlert.m
@@ -1,3 +1,4 @@
+#import "UIKitPrivate.h"
 #import "UIViewController+LCAlert.h"
 
 @implementation UIViewController(LCAlert)