1 files changed, 2 insertions, 1 deletions

diff --git a/arch/x86/mm/init_32.c b/arch/x86/mm/init_32.c
index e39504878aec..7d05565ba781 100644
--- a/arch/x86/mm/init_32.c
+++ b/arch/x86/mm/init_32.c
@@ -825,7 +825,8 @@ void __init mem_init(void)
 int arch_add_memory(int nid, u64 start, u64 size)
 {
         struct pglist_data *pgdata = NODE_DATA(nid);
-        struct zone *zone = pgdata->node_zones + ZONE_HIGHMEM;
+        struct zone *zone = pgdata->node_zones +
+                zone_for_memory(nid, start, size, ZONE_HIGHMEM);
         unsigned long start_pfn = start >> PAGE_SHIFT;
         unsigned long nr_pages = size >> PAGE_SHIFT;