#define UPDATE_DT_NODE 0x02000000
 #define ADD_DT_NODE    0x03000000
 
-static int mobility_rtas_call(int token, char *buf)
+#define MIGRATION_SCOPE        (1)
+
+static int mobility_rtas_call(int token, char *buf, s32 scope)
 {
        int rc;
 
        spin_lock(&rtas_data_buf_lock);
 
        memcpy(rtas_data_buf, buf, RTAS_DATA_BUF_SIZE);
-       rc = rtas_call(token, 2, 1, NULL, rtas_data_buf, 1);
+       rc = rtas_call(token, 2, 1, NULL, rtas_data_buf, scope);
        memcpy(buf, rtas_data_buf, RTAS_DATA_BUF_SIZE);
 
        spin_unlock(&rtas_data_buf_lock);
        return 0;
 }
 
-static int update_dt_node(u32 phandle)
+static int update_dt_node(u32 phandle, s32 scope)
 {
        struct update_props_workarea *upwa;
        struct device_node *dn;
        upwa->phandle = phandle;
 
        do {
-               rc = mobility_rtas_call(update_properties_token, rtas_buf);
+               rc = mobility_rtas_call(update_properties_token, rtas_buf,
+                                       scope);
                if (rc < 0)
                        break;
 
        return rc;
 }
 
-static int pseries_devicetree_update(void)
+int pseries_devicetree_update(s32 scope)
 {
        char *rtas_buf;
        u32 *data;
                return -ENOMEM;
 
        do {
-               rc = mobility_rtas_call(update_nodes_token, rtas_buf);
+               rc = mobility_rtas_call(update_nodes_token, rtas_buf, scope);
                if (rc && rc != 1)
                        break;
 
                                        delete_dt_node(phandle);
                                        break;
                                case UPDATE_DT_NODE:
-                                       update_dt_node(phandle);
+                                       update_dt_node(phandle, scope);
                                        break;
                                case ADD_DT_NODE:
                                        drc_index = *data++;
        int rc;
        int activate_fw_token;
 
-       rc = pseries_devicetree_update();
+       rc = pseries_devicetree_update(MIGRATION_SCOPE);
        if (rc) {
                printk(KERN_ERR "Initial post-mobility device tree update "
                       "failed: %d\n", rc);
 
        rc = rtas_call(activate_fw_token, 0, 1, NULL);
        if (!rc) {
-               rc = pseries_devicetree_update();
+               rc = pseries_devicetree_update(MIGRATION_SCOPE);
                if (rc)
                        printk(KERN_ERR "Secondary post-mobility device tree "
                               "update failed: %d\n", rc);