of_scan_flat_dt(early_init_dt_scan_chosen_ppc, boot_command_line);
 
        /* Scan memory nodes and rebuild MEMBLOCKs */
-       of_scan_flat_dt(early_init_dt_scan_root, NULL);
+       early_init_dt_scan_root();
        of_scan_flat_dt(early_init_dt_scan_memory_ppc, NULL);
 
        parse_early_param();
         * mess the memblock.
         */
        add_mem_to_memblock = 0;
-       of_scan_flat_dt(early_init_dt_scan_root, NULL);
+       early_init_dt_scan_root();
        of_scan_flat_dt(early_init_dt_scan_memory_ppc, NULL);
        add_mem_to_memblock = 1;
 
 
 /*
  * early_init_dt_scan_root - fetch the top level address and size cells
  */
-int __init early_init_dt_scan_root(unsigned long node, const char *uname,
-                                  int depth, void *data)
+int __init early_init_dt_scan_root(void)
 {
        const __be32 *prop;
+       const void *fdt = initial_boot_params;
+       int node = fdt_path_offset(fdt, "/");
 
-       if (depth != 0)
-               return 0;
+       if (node < 0)
+               return -ENODEV;
 
        dt_root_size_cells = OF_ROOT_NODE_SIZE_CELLS_DEFAULT;
        dt_root_addr_cells = OF_ROOT_NODE_ADDR_CELLS_DEFAULT;
                dt_root_addr_cells = be32_to_cpup(prop);
        pr_debug("dt_root_addr_cells = %x\n", dt_root_addr_cells);
 
-       /* break now */
-       return 1;
+       return 0;
 }
 
 u64 __init dt_mem_next_cell(int s, const __be32 **cellp)
        int rc;
 
        /* Initialize {size,address}-cells info */
-       of_scan_flat_dt(early_init_dt_scan_root, NULL);
+       early_init_dt_scan_root();
 
        /* Retrieve various information from the /chosen node */
        rc = early_init_dt_scan_chosen(boot_command_line);
 
 extern u64 dt_mem_next_cell(int s, const __be32 **cellp);
 
 /* Early flat tree scan hooks */
-extern int early_init_dt_scan_root(unsigned long node, const char *uname,
-                                  int depth, void *data);
+extern int early_init_dt_scan_root(void);
 
 extern bool early_init_dt_scan(void *params);
 extern bool early_init_dt_verify(void *params);