return ret;
 }
 
+static void gpio_set_low_action(void *data)
+{
+       struct lm3532_data *priv = data;
+
+       gpiod_direction_output(priv->enable_gpio, 0);
+}
+
 static int lm3532_parse_node(struct lm3532_data *priv)
 {
        struct fwnode_handle *child = NULL;
        if (IS_ERR(priv->enable_gpio))
                priv->enable_gpio = NULL;
 
+       if (priv->enable_gpio) {
+               ret = devm_add_action(&priv->client->dev, gpio_set_low_action, priv);
+               if (ret)
+                       return ret;
+       }
+
        priv->regulator = devm_regulator_get(&priv->client->dev, "vin");
        if (IS_ERR(priv->regulator))
                priv->regulator = NULL;
                return ret;
        }
 
-       mutex_init(&drvdata->lock);
+       ret = devm_mutex_init(&client->dev, &drvdata->lock);
+       if (ret)
+               return ret;
+
        i2c_set_clientdata(client, drvdata);
 
        ret = lm3532_parse_node(drvdata);
        return ret;
 }
 
-static void lm3532_remove(struct i2c_client *client)
-{
-       struct lm3532_data *drvdata = i2c_get_clientdata(client);
-
-       mutex_destroy(&drvdata->lock);
-
-       if (drvdata->enable_gpio)
-               gpiod_direction_output(drvdata->enable_gpio, 0);
-}
-
 static const struct of_device_id of_lm3532_leds_match[] = {
        { .compatible = "ti,lm3532", },
        {},
 
 static struct i2c_driver lm3532_i2c_driver = {
        .probe = lm3532_probe,
-       .remove = lm3532_remove,
        .id_table = lm3532_id,
        .driver = {
                .name = LM3532_NAME,