1 files changed, 6 insertions, 0 deletions

diff --git a/device.h b/device.h
index 8788b2d14..9510fbedb 100644
--- a/device.h
+++ b/device.h
@@ -58,6 +58,12 @@ class Device {
     return ui_;
   }
 
+  // Sets the UI object to the given UI. Used to override the default UI in case initialization
+  // failed, or we want a stub for some reason.
+  virtual void SetUI(RecoveryUI* ui) {
+    ui_ = ui;
+  }
+
   // Called when recovery starts up (after the UI has been obtained and initialized and after the
   // arguments have been parsed, but before anything else).
   virtual void StartRecovery() {};