diff --git a/mock_payload_state.h b/mock_payload_state.h
index 88578b5..65d99da 100644
--- a/mock_payload_state.h
+++ b/mock_payload_state.h
@@ -24,7 +24,7 @@
   MOCK_METHOD0(UpdateResumed, void());
   MOCK_METHOD0(UpdateRestarted, void());
   MOCK_METHOD0(UpdateSucceeded, void());
-  MOCK_METHOD1(UpdateFailed, void(ActionExitCode error));
+  MOCK_METHOD1(UpdateFailed, void(ErrorCode error));
   MOCK_METHOD0(ShouldBackoffDownload, bool());
 
   // Getters.