diff --git a/Classes/BITUpdateViewControllerPrivate.h b/Classes/BITUpdateViewControllerPrivate.h
index 54aa07952c..310d10c500 100644
--- a/Classes/BITUpdateViewControllerPrivate.h
+++ b/Classes/BITUpdateViewControllerPrivate.h
@@ -29,6 +29,7 @@
  */
 
 #import <UIKit/UIKit.h>
+#import "PSStoreButton.h"
 
 typedef enum {
 	AppStoreButtonStateOffline,