diff --git a/mock_payload_state.h b/mock_payload_state.h
index 65d99da..6513a83 100644
--- a/mock_payload_state.h
+++ b/mock_payload_state.h
@@ -30,7 +30,7 @@
   // Getters.
   MOCK_METHOD0(GetResponseSignature, std::string());
   MOCK_METHOD0(GetPayloadAttemptNumber, uint32_t());
-  MOCK_METHOD0(GetUrlIndex, uint32_t());
+  MOCK_METHOD0(GetCurrentUrl, std::string());
   MOCK_METHOD0(GetUrlFailureCount, uint32_t());
   MOCK_METHOD0(GetUrlSwitchCount, uint32_t());
   MOCK_METHOD0(GetBackoffExpiryTime, base::Time());