diff --git a/arch/i386/mm/highmem.c b/arch/i386/mm/highmem.c
index a1a21abf742ebc9b4926ab4bf334da2d59cf37c9..ad8d86cc683ee3f6786142c86a95f1fffb7e4f94 100644
--- a/arch/i386/mm/highmem.c
+++ b/arch/i386/mm/highmem.c
@@ -72,6 +72,7 @@ void kunmap_atomic(void *kvaddr, enum km_type type)
 #endif
 	}
 
+	arch_flush_lazy_mmu_mode();
 	pagefault_enable();
 }