1 files changed, 3 insertions, 5 deletions

diff --git a/drivers/mtd/maps/physmap_of.c b/drivers/mtd/maps/physmap_of.c
index 263d9e1b8001..500d7f4f8028 100644
--- a/drivers/mtd/maps/physmap_of.c
+++ b/drivers/mtd/maps/physmap_of.c
@@ -68,9 +68,6 @@ static int of_flash_remove(struct platform_device *dev)
                         kfree(info->list[i].res);
                 }
         }
-
-        kfree(info);
-
         return 0;
 }
 
@@ -199,8 +196,9 @@ static int of_flash_probe(struct platform_device *dev)
         map_indirect = of_property_read_bool(dp, "no-unaligned-direct-access");
 
         err = -ENOMEM;
-        info = kzalloc(sizeof(struct of_flash) +
-                       sizeof(struct of_flash_list) * count, GFP_KERNEL);
+        info = devm_kzalloc(&dev->dev,
+                            sizeof(struct of_flash) +
+                            sizeof(struct of_flash_list) * count, GFP_KERNEL);
         if (!info)
                 goto err_flash_remove;