diff --git a/payload_state.h b/payload_state.h
index b12e6b0..1b0899c 100644
--- a/payload_state.h
+++ b/payload_state.h
@@ -38,7 +38,7 @@
   virtual void UpdateResumed();
   virtual void UpdateRestarted();
   virtual void UpdateSucceeded();
-  virtual void UpdateFailed(ActionExitCode error);
+  virtual void UpdateFailed(ErrorCode error);
   virtual bool ShouldBackoffDownload();
 
   virtual inline std::string GetResponseSignature() {