1 files changed, 2 insertions, 2 deletions

diff --git a/arch/um/kernel/mem.c b/arch/um/kernel/mem.c
index 462cc9d65386..fa4f915be5c5 100644
--- a/arch/um/kernel/mem.c
+++ b/arch/um/kernel/mem.c
@@ -234,8 +234,8 @@ void paging_init(void)
         empty_bad_page = (unsigned long *) alloc_bootmem_low_pages(PAGE_SIZE);
         for(i=0;i<sizeof(zones_size)/sizeof(zones_size[0]);i++) 
                 zones_size[i] = 0;
-        zones_size[0] = (end_iomem >> PAGE_SHIFT) - (uml_physmem >> PAGE_SHIFT);
-        zones_size[2] = highmem >> PAGE_SHIFT;
+        zones_size[ZONE_DMA] = (end_iomem >> PAGE_SHIFT) - (uml_physmem >> PAGE_SHIFT);
+        zones_size[ZONE_HIGHMEM] = highmem >> PAGE_SHIFT;
         free_area_init(zones_size);
 
         /*