diff --git a/mock_update_attempter.h b/mock_update_attempter.h
index 9634bab..d64dcbe 100644
--- a/mock_update_attempter.h
+++ b/mock_update_attempter.h
@@ -42,6 +42,8 @@
 
   MOCK_METHOD0(ResetStatus, bool(void));
 
+  MOCK_METHOD0(GetCurrentUpdateAttemptFlags, UpdateAttemptFlags(void));
+
   MOCK_METHOD3(CheckForUpdate, void(const std::string& app_version,
                                     const std::string& omaha_url,
                                     bool is_interactive));