#include <linux/interrupt.h>
 #include <linux/regmap.h>
 #include <linux/bitops.h>
+#include <linux/devm-helpers.h>
 
 #define SBS_CHARGER_REG_SPEC_INFO              0x11
 #define SBS_CHARGER_REG_STATUS                 0x13
                if (ret)
                        return dev_err_probe(&client->dev, ret, "Failed to request irq\n");
        } else {
-               INIT_DELAYED_WORK(&chip->work, sbs_delayed_work);
+               ret = devm_delayed_work_autocancel(&client->dev, &chip->work,
+                                                  sbs_delayed_work);
+               if (ret)
+                       return dev_err_probe(&client->dev, ret,
+                                            "Failed to init work for polling\n");
+
                schedule_delayed_work(&chip->work,
                                      msecs_to_jiffies(SBS_CHARGER_POLL_TIME));
        }
        return 0;
 }
 
-static int sbs_remove(struct i2c_client *client)
-{
-       struct sbs_info *chip = i2c_get_clientdata(client);
-
-       cancel_delayed_work_sync(&chip->work);
-
-       return 0;
-}
-
 #ifdef CONFIG_OF
 static const struct of_device_id sbs_dt_ids[] = {
        { .compatible = "sbs,sbs-charger" },
 
 static struct i2c_driver sbs_driver = {
        .probe          = sbs_probe,
-       .remove         = sbs_remove,
        .id_table       = sbs_id,
        .driver = {
                .name   = "sbs-charger",