return -ENOMEM;
 
        aic31xx->regmap = devm_regmap_init_i2c(i2c, regmap_config);
-
        if (IS_ERR(aic31xx->regmap)) {
                ret = PTR_ERR(aic31xx->regmap);
                dev_err(&i2c->dev, "Failed to allocate register map: %d\n",
 
        aic31xx_device_init(aic31xx);
 
-       ret = snd_soc_register_codec(&i2c->dev, &soc_codec_driver_aic31xx,
+       return snd_soc_register_codec(&i2c->dev, &soc_codec_driver_aic31xx,
                                     aic31xx_dai_driver,
                                     ARRAY_SIZE(aic31xx_dai_driver));
-
-       return ret;
 }
 
 static int aic31xx_i2c_remove(struct i2c_client *i2c)
 {
-       struct aic31xx_priv *aic31xx = dev_get_drvdata(&i2c->dev);
-
-       kfree(aic31xx);
+       snd_soc_unregister_codec(&i2c->dev);
        return 0;
 }
 
                .of_match_table = of_match_ptr(tlv320aic31xx_of_match),
        },
        .probe          = aic31xx_i2c_probe,
-       .remove         = (aic31xx_i2c_remove),
+       .remove         = aic31xx_i2c_remove,
        .id_table       = aic31xx_i2c_id,
 };