diff --git a/device.cpp b/device.cpp
index f881daf..3b0942c 100644
--- a/device.cpp
+++ b/device.cpp
@@ -16,6 +16,8 @@
 
 #include "device.h"
 
+#include "ui.h"
+
 static const char* MENU_ITEMS[] = {
   "Reboot system now",
   "Reboot to bootloader",
diff --git a/device.h b/device.h
index 74745b3..4ea3159 100644
--- a/device.h
+++ b/device.h
@@ -17,7 +17,8 @@
 #ifndef _RECOVERY_DEVICE_H
 #define _RECOVERY_DEVICE_H
 
-#include "ui.h"
+// Forward declaration to avoid including "ui.h".
+class RecoveryUI;
 
 class Device {
  public: