diff --git a/payload_state_interface.h b/payload_state_interface.h
index 3adc148..d384a0e 100644
--- a/payload_state_interface.h
+++ b/payload_state_interface.h
@@ -205,6 +205,9 @@
 
   // Switch to next payload.
   virtual bool NextPayload() = 0;
+
+  // Sets and persists the staging wallclock-based wait period.
+  virtual void SetStagingWaitPeriod(base::TimeDelta wait_period) = 0;
 };
 
 }  // namespace chromeos_update_engine