diff --git a/update_attempter.h b/update_attempter.h
index abd0bd4..6c93150 100644
--- a/update_attempter.h
+++ b/update_attempter.h
@@ -90,6 +90,7 @@
                       bool rollback_allowed,
                       bool rollback_data_save_requested,
                       int rollback_allowed_milestones,
+                      bool rollback_on_channel_downgrade,
                       bool obey_proxies,
                       bool interactive);
 
@@ -284,6 +285,8 @@
   FRIEND_TEST(UpdateAttempterTest, RollbackAfterInstall);
   FRIEND_TEST(UpdateAttempterTest, RollbackAllowed);
   FRIEND_TEST(UpdateAttempterTest, RollbackAllowedSetAndReset);
+  FRIEND_TEST(UpdateAttempterTest, ChannelDowngradeNoRollback);
+  FRIEND_TEST(UpdateAttempterTest, ChannelDowngradeRollback);
   FRIEND_TEST(UpdateAttempterTest, RollbackMetricsNotRollbackFailure);
   FRIEND_TEST(UpdateAttempterTest, RollbackMetricsNotRollbackSuccess);
   FRIEND_TEST(UpdateAttempterTest, RollbackMetricsRollbackFailure);
@@ -376,6 +379,7 @@
                              bool rollback_allowed,
                              bool rollback_data_save_requested,
                              int rollback_allowed_milestones,
+                             bool rollback_on_channel_downgrade,
                              bool obey_proxies,
                              bool interactive);