1 files changed, 3 insertions, 4 deletions

diff --git a/drivers/regulator/core.c b/drivers/regulator/core.c
index f98a8ee29933..974276afd1f0 100644
--- a/drivers/regulator/core.c
+++ b/drivers/regulator/core.c
@@ -1063,6 +1063,8 @@ static struct regulator *create_regulator(struct regulator_dev *rdev,
         list_add(&regulator->list, &rdev->consumer_list);
 
         if (dev) {
+                regulator->dev = dev;
+
                 /* Add a link to the device sysfs entry */
                 size = scnprintf(buf, REG_STR_SIZE, "%s-%s",
                                  dev->kobj.name, supply_name);
@@ -1359,11 +1361,8 @@ void regulator_put(struct regulator *regulator)
         debugfs_remove_recursive(regulator->debugfs);
 
         /* remove any sysfs entries */
-        if (regulator->dev) {
+        if (regulator->dev)
                 sysfs_remove_link(&rdev->dev.kobj, regulator->supply_name);
-                device_remove_file(regulator->dev, &regulator->dev_attr);
-                kfree(regulator->dev_attr.attr.name);
-        }
         kfree(regulator->supply_name);
         list_del(&regulator->list);
         kfree(regulator);