diff --git a/src/displayapp/DisplayApp.h b/src/displayapp/DisplayApp.h
index 9009d5daee..0be91e848a 100644
--- a/src/displayapp/DisplayApp.h
+++ b/src/displayapp/DisplayApp.h
@@ -14,6 +14,7 @@
 #include "displayapp/screens/Screen.h"
 #include "components/timer/Timer.h"
 #include "components/alarm/AlarmController.h"
+#include "components/infinisleep/InfiniSleepController.h"
 #include "touchhandler/TouchHandler.h"
 
 #include "displayapp/Messages.h"
diff --git a/src/displayapp/DisplayAppRecovery.h b/src/displayapp/DisplayAppRecovery.h
index 6cb3e7394d..dc5108b66f 100644
--- a/src/displayapp/DisplayAppRecovery.h
+++ b/src/displayapp/DisplayAppRecovery.h
@@ -37,6 +37,7 @@ namespace Pinetime {
     class SimpleWeatherService;
     class MusicService;
     class NavigationService;
+    class InfiniSleepController;
   }
 
   namespace System {