lpc18xx_otp_nvmem_config.dev = &pdev->dev;
        lpc18xx_otp_nvmem_config.priv = otp;
 
-       nvmem = nvmem_register(&lpc18xx_otp_nvmem_config);
-       if (IS_ERR(nvmem))
-               return PTR_ERR(nvmem);
+       nvmem = devm_nvmem_register(&pdev->dev, &lpc18xx_otp_nvmem_config);
 
-       platform_set_drvdata(pdev, nvmem);
-
-       return 0;
-}
-
-static int lpc18xx_otp_remove(struct platform_device *pdev)
-{
-       struct nvmem_device *nvmem = platform_get_drvdata(pdev);
-
-       return nvmem_unregister(nvmem);
+       return PTR_ERR_OR_ZERO(nvmem);
 }
 
 static const struct of_device_id lpc18xx_otp_dt_ids[] = {
 
 static struct platform_driver lpc18xx_otp_driver = {
        .probe  = lpc18xx_otp_probe,
-       .remove = lpc18xx_otp_remove,
        .driver = {
                .name   = "lpc18xx_otp",
                .of_match_table = lpc18xx_otp_dt_ids,