1 files changed, 3 insertions, 3 deletions

diff --git a/arch/x86/kernel/efi.c b/arch/x86/kernel/efi.c
index 32dd62b36ff7..b4d523276f40 100644
--- a/arch/x86/kernel/efi.c
+++ b/arch/x86/kernel/efi.c
@@ -428,9 +428,6 @@ void __init efi_enter_virtual_mode(void)
                 else
                         va = efi_ioremap(md->phys_addr, size);
 
-                if (md->attribute & EFI_MEMORY_WB)
-                        set_memory_uc(md->virt_addr, size);
-
                 md->virt_addr = (u64) (unsigned long) va;
 
                 if (!va) {
@@ -439,6 +436,9 @@ void __init efi_enter_virtual_mode(void)
                         continue;
                 }
 
+                if (!(md->attribute & EFI_MEMORY_WB))
+                        set_memory_uc(md->virt_addr, size);
+
                 systab = (u64) (unsigned long) efi_phys.systab;
                 if (md->phys_addr <= systab && systab < end) {
                         systab += md->virt_addr - md->phys_addr;