diff --git a/mock_update_attempter.h b/mock_update_attempter.h
index d97163d..c39fb62 100644
--- a/mock_update_attempter.h
+++ b/mock_update_attempter.h
@@ -30,12 +30,13 @@
  public:
   using UpdateAttempter::UpdateAttempter;
 
-  MOCK_METHOD8(Update,
+  MOCK_METHOD9(Update,
                void(const std::string& app_version,
                     const std::string& omaha_url,
                     const std::string& target_channel,
                     const std::string& target_version_prefix,
                     bool rollback_allowed,
+                    bool rollback_data_save_requested,
                     int rollback_allowed_milestones,
                     bool obey_proxies,
                     bool interactive));