1 files changed, 4 insertions, 2 deletions

diff --git a/drivers/firmware/efi/libstub/efi-stub-helper.c b/drivers/firmware/efi/libstub/efi-stub-helper.c
index 29ed2f9b218c..3bd127f95315 100644
--- a/drivers/firmware/efi/libstub/efi-stub-helper.c
+++ b/drivers/firmware/efi/libstub/efi-stub-helper.c
@@ -125,10 +125,12 @@ unsigned long get_dram_base(efi_system_table_t *sys_table_arg)
 
         map.map_end = map.map + map_size;
 
-        for_each_efi_memory_desc(&map, md)
-                if (md->attribute & EFI_MEMORY_WB)
+        for_each_efi_memory_desc_in_map(&map, md) {
+                if (md->attribute & EFI_MEMORY_WB) {
                         if (membase > md->phys_addr)
                                 membase = md->phys_addr;
+                }
+        }
 
         efi_call_early(free_pool, map.map);