/**
  * pm_genpd_remove_device - Remove a device from an I/O PM domain.
- * @genpd: PM domain to remove the device from.
  * @dev: Device to be removed.
  */
-int pm_genpd_remove_device(struct generic_pm_domain *genpd,
-                          struct device *dev)
+int pm_genpd_remove_device(struct device *dev)
 {
-       if (!genpd || genpd != genpd_lookup_dev(dev))
+       struct generic_pm_domain *genpd = genpd_lookup_dev(dev);
+
+       if (!genpd)
                return -EINVAL;
 
        return genpd_remove_device(genpd, dev);
 
        if (adev->acp.acp_genpd) {
                for (i = 0; i < ACP_DEVS ; i++) {
                        dev = get_mfd_cell_dev(adev->acp.acp_cell[i].name, i);
-                       ret = pm_genpd_remove_device(&adev->acp.acp_genpd->gpd, dev);
+                       ret = pm_genpd_remove_device(dev);
                        /* If removal fails, dont giveup and try rest */
                        if (ret)
                                dev_err(dev, "remove dev from genpd failed\n");
 
 }
 
 int pm_genpd_add_device(struct generic_pm_domain *genpd, struct device *dev);
-int pm_genpd_remove_device(struct generic_pm_domain *genpd, struct device *dev);
+int pm_genpd_remove_device(struct device *dev);
 int pm_genpd_add_subdomain(struct generic_pm_domain *genpd,
                           struct generic_pm_domain *new_subdomain);
 int pm_genpd_remove_subdomain(struct generic_pm_domain *genpd,
 {
        return -ENOSYS;
 }
-static inline int pm_genpd_remove_device(struct generic_pm_domain *genpd,
-                                        struct device *dev)
+static inline int pm_genpd_remove_device(struct device *dev)
 {
        return -ENOSYS;
 }