1 files changed, 1 insertions, 6 deletions

diff --git a/drivers/mtd/nand/nomadik_nand.c b/drivers/mtd/nand/nomadik_nand.c
index a045a4a581b6..b6a5c86ab31e 100644
--- a/drivers/mtd/nand/nomadik_nand.c
+++ b/drivers/mtd/nand/nomadik_nand.c
@@ -158,12 +158,7 @@ static int nomadik_nand_probe(struct platform_device *pdev)
                 goto err_unmap;
         }
 
-#ifdef CONFIG_MTD_PARTITIONS
-        add_mtd_partitions(&host->mtd, pdata->parts, pdata->nparts);
-#else
-        pr_info("Registering %s as whole device\n", mtd->name);
-        add_mtd_device(mtd);
-#endif
+        mtd_device_register(&host->mtd, pdata->parts, pdata->nparts);
 
         platform_set_drvdata(pdev, host);
         return 0;