void __init plat_mem_setup(void)
 {
+       unsigned int nid = 1;
+
        /* Register CPU#0 URAM space as Node 1 */
-       setup_bootmem_node(1, 0x145f0000, 0x14610000);  /* CPU0 */
+       setup_bootmem_node(nid++, 0x145f0000, 0x14610000);      /* CPU0 */
 
 #if 0
        /* XXX: Not yet.. */
-       setup_bootmem_node(2, 0x14df0000, 0x14e10000);  /* CPU1 */
-       setup_bootmem_node(3, 0x155f0000, 0x15610000);  /* CPU2 */
-       setup_bootmem_node(4, 0x15df0000, 0x15e10000);  /* CPU3 */
-       setup_bootmem_node(5, 0x16000000, 0x16020000);  /* CSM */
+       setup_bootmem_node(nid++, 0x14df0000, 0x14e10000);      /* CPU1 */
+       setup_bootmem_node(nid++, 0x155f0000, 0x15610000);      /* CPU2 */
+       setup_bootmem_node(nid++, 0x15df0000, 0x15e10000);      /* CPU3 */
 #endif
+
+       setup_bootmem_node(nid++, 0x16000000, 0x16020000);      /* CSM */
 }