da9063->irq_base = regmap_irq_chip_get_base(da9063->regmap_irq);
 
-       ret = mfd_add_devices(da9063->dev, -1, da9063_devs,
-                             ARRAY_SIZE(da9063_devs), NULL, da9063->irq_base,
-                             NULL);
+       ret = devm_mfd_add_devices(da9063->dev, -1, da9063_devs,
+                                  ARRAY_SIZE(da9063_devs), NULL,
+                                  da9063->irq_base, NULL);
        if (ret)
                dev_err(da9063->dev, "Cannot add MFD cells\n");
 
        return ret;
 }
 
-void da9063_device_exit(struct da9063 *da9063)
-{
-       mfd_remove_devices(da9063->dev);
-}
-
 MODULE_DESCRIPTION("PMIC driver for Dialog DA9063");
 MODULE_AUTHOR("Krystian Garbaciak");
 MODULE_AUTHOR("Michal Hajduk");
 
        return da9063_device_init(da9063, i2c->irq);
 }
 
-static int da9063_i2c_remove(struct i2c_client *i2c)
-{
-       struct da9063 *da9063 = i2c_get_clientdata(i2c);
-
-       da9063_device_exit(da9063);
-
-       return 0;
-}
-
 static const struct i2c_device_id da9063_i2c_id[] = {
        {"da9063", PMIC_DA9063},
        {},
                .of_match_table = of_match_ptr(da9063_dt_ids),
        },
        .probe    = da9063_i2c_probe,
-       .remove   = da9063_i2c_remove,
        .id_table = da9063_i2c_id,
 };
 
 
 int da9063_device_init(struct da9063 *da9063, unsigned int irq);
 int da9063_irq_init(struct da9063 *da9063);
 
-void da9063_device_exit(struct da9063 *da9063);
-
 #endif /* __MFD_DA9063_CORE_H__ */