1 files changed, 3 insertions, 9 deletions

diff --git a/drivers/mtd/nand/tmio_nand.c b/drivers/mtd/nand/tmio_nand.c
index b6ffad64cda..beebd95f769 100644
--- a/drivers/mtd/nand/tmio_nand.c
+++ b/drivers/mtd/nand/tmio_nand.c
@@ -378,8 +378,6 @@ static int tmio_probe(struct platform_device *dev)
         struct tmio_nand *tmio;
         struct mtd_info *mtd;
         struct nand_chip *nand_chip;
-        struct mtd_partition *parts;
-        int nbparts = 0;
         int retval;
 
         if (data == NULL)
@@ -458,13 +456,9 @@ static int tmio_probe(struct platform_device *dev)
                 goto err_scan;
         }
         /* Register the partitions */
-        nbparts = parse_mtd_partitions(mtd, NULL, &parts, 0);
-        if (nbparts <= 0 && data) {
-                parts = data->partition;
-                nbparts = data->num_partitions;
-        }
-
-        retval = mtd_device_register(mtd, parts, nbparts);
+        retval = mtd_device_parse_register(mtd, NULL, 0,
+                        data ? data->partition : NULL,
+                        data ? data->num_partitions : 0);
         if (!retval)
                 return retval;