diff --git a/binder_service_brillo.h b/binder_service_brillo.h
index 51f0fab..81a4e4c 100644
--- a/binder_service_brillo.h
+++ b/binder_service_brillo.h
@@ -51,6 +51,7 @@
   void SendPayloadApplicationComplete(ErrorCode error_code) override {}
 
   // android::brillo::BnUpdateEngine overrides.
+  android::binder::Status SetUpdateAttemptFlags(int flags) override;
   android::binder::Status AttemptUpdate(const android::String16& app_version,
                                         const android::String16& omaha_url,
                                         int flags) override;