return policy;
 }
 
-static struct cpufreq_policy *cpufreq_policy_alloc(int cpu)
+static struct cpufreq_policy *cpufreq_policy_alloc(struct device *dev)
 {
        struct cpufreq_policy *policy;
+       int ret;
 
        policy = kzalloc(sizeof(*policy), GFP_KERNEL);
        if (!policy)
        if (!zalloc_cpumask_var(&policy->related_cpus, GFP_KERNEL))
                goto err_free_cpumask;
 
+       ret = kobject_init_and_add(&policy->kobj, &ktype_cpufreq, &dev->kobj,
+                                  "cpufreq");
+       if (ret) {
+               pr_err("%s: failed to init policy->kobj: %d\n", __func__, ret);
+               goto err_free_rcpumask;
+       }
+
        INIT_LIST_HEAD(&policy->policy_list);
        init_rwsem(&policy->rwsem);
        spin_lock_init(&policy->transition_lock);
        init_completion(&policy->kobj_unregister);
        INIT_WORK(&policy->update, handle_update);
 
-       policy->cpu = cpu;
+       policy->cpu = dev->id;
 
        /* Set this once on allocation */
-       policy->kobj_cpu = cpu;
+       policy->kobj_cpu = dev->id;
 
        return policy;
 
+err_free_rcpumask:
+       free_cpumask_var(policy->related_cpus);
 err_free_cpumask:
        free_cpumask_var(policy->cpus);
 err_free_policy:
        return NULL;
 }
 
-static void cpufreq_policy_put_kobj(struct cpufreq_policy *policy)
+static void cpufreq_policy_put_kobj(struct cpufreq_policy *policy, bool notify)
 {
        struct kobject *kobj;
        struct completion *cmp;
 
-       blocking_notifier_call_chain(&cpufreq_policy_notifier_list,
-                       CPUFREQ_REMOVE_POLICY, policy);
+       if (notify)
+               blocking_notifier_call_chain(&cpufreq_policy_notifier_list,
+                                            CPUFREQ_REMOVE_POLICY, policy);
 
        down_write(&policy->rwsem);
        cpufreq_remove_dev_symlink(policy);
        policy = recover_policy ? cpufreq_policy_restore(cpu) : NULL;
        if (!policy) {
                recover_policy = false;
-               policy = cpufreq_policy_alloc(cpu);
+               policy = cpufreq_policy_alloc(dev);
                if (!policy)
                        goto nomem_out;
        }
                policy->user_policy.min = policy->min;
                policy->user_policy.max = policy->max;
 
-               /* prepare interface data */
-               ret = kobject_init_and_add(&policy->kobj, &ktype_cpufreq,
-                                          &dev->kobj, "cpufreq");
-               if (ret) {
-                       pr_err("%s: failed to init policy->kobj: %d\n",
-                              __func__, ret);
-                       goto err_init_policy_kobj;
-               }
-
                write_lock_irqsave(&cpufreq_driver_lock, flags);
                for_each_cpu(j, policy->related_cpus)
                        per_cpu(cpufreq_cpu_data, j) = policy;
 
 err_out_unregister:
 err_get_freq:
-       if (!recover_policy) {
-               kobject_put(&policy->kobj);
-               wait_for_completion(&policy->kobj_unregister);
-       }
-err_init_policy_kobj:
        up_write(&policy->rwsem);
 
        if (cpufreq_driver->exit)
                cpufreq_driver->exit(policy);
 err_set_policy_cpu:
-       if (recover_policy)
-               cpufreq_policy_put_kobj(policy);
+       cpufreq_policy_put_kobj(policy, recover_policy);
        cpufreq_policy_free(policy);
 
 nomem_out:
 
        /* Free the policy kobjects only if the driver is getting removed. */
        if (sif)
-               cpufreq_policy_put_kobj(policy);
+               cpufreq_policy_put_kobj(policy, true);
 
        /*
         * Perform the ->exit() even during light-weight tear-down,
                        return 0;
                }
 
-               cpufreq_policy_put_kobj(policy);
+               cpufreq_policy_put_kobj(policy, true);
                cpufreq_policy_free(policy);
                return 0;
        }