diff --git a/firmware/include/vboot_api.h b/firmware/include/vboot_api.h
index 4b37a07b72..3f0a7faf16 100644
--- a/firmware/include/vboot_api.h
+++ b/firmware/include/vboot_api.h
@@ -636,8 +636,10 @@ VbError_t VbExEcStayInRO(void);
 /* Read the SHA-256 hash of the rewriteable EC image. */
 VbError_t VbExEcHashRW(const uint8_t **hash, int *hash_size);
 
-/* Get the expected contents of the EC image as stored in the main firmware. */
-VbError_t VbExEcGetExpectedRW(const uint8_t **image, int *image_size);
+/* Get the expected contents of the EC image associated with the main firmware
+ * specified by the "select" argument. */
+VbError_t VbExEcGetExpectedRW(enum VbSelectFirmware_t select, uint8_t **image,
+                              int *image_size);
 
 /* Update the EC rewritable image. */
 VbError_t VbExEcUpdateRW(const uint8_t *image, int image_size);
diff --git a/firmware/stub/vboot_api_stub.c b/firmware/stub/vboot_api_stub.c
index 6452e97d81..f41d8d9573 100644
--- a/firmware/stub/vboot_api_stub.c
+++ b/firmware/stub/vboot_api_stub.c
@@ -173,8 +173,9 @@ VbError_t VbExEcHashRW(const uint8_t **hash, int *hash_size) {
   return VBERROR_SUCCESS;
 }
 
-VbError_t VbExEcGetExpectedRW(const uint8_t **image, int *image_size) {
-  static const uint8_t fake_image[64] = {5, 6, 7, 8};
+VbError_t VbExEcGetExpectedRW(enum VbSelectFirmware_t select, uint8_t **image,
+                              int *image_size) {
+  static uint8_t fake_image[64] = {5, 6, 7, 8};
   *image = fake_image;
   *image_size = sizeof(fake_image);
   return VBERROR_SUCCESS;