diff --git a/mock_service_observer.h b/mock_service_observer.h
index 032f55a..e434eab 100644
--- a/mock_service_observer.h
+++ b/mock_service_observer.h
@@ -24,12 +24,9 @@
 
 class MockServiceObserver : public ServiceObserverInterface {
  public:
-  MOCK_METHOD5(SendStatusUpdate,
-               void(int64_t last_checked_time,
-                    double progress,
-                    update_engine::UpdateStatus status,
-                    const std::string& new_version,
-                    int64_t new_size));
+  MOCK_METHOD1(
+      SendStatusUpdate,
+      void(const update_engine::UpdateEngineStatus& update_engine_status));
   MOCK_METHOD1(SendPayloadApplicationComplete, void(ErrorCode error_code));
 };