1 files changed, 3 insertions, 1 deletions

diff --git a/drivers/char/tpm/tpm-chip.c b/drivers/char/tpm/tpm-chip.c
index 1a9dcee8da5a..252142524ff2 100644
--- a/drivers/char/tpm/tpm-chip.c
+++ b/drivers/char/tpm/tpm-chip.c
@@ -136,6 +136,8 @@ struct tpm_chip *tpmm_chip_alloc(struct device *dev,
         chip->cdev.owner = chip->pdev->driver->owner;
         chip->cdev.kobj.parent = &chip->dev.kobj;
 
+        devm_add_action(dev, (void (*)(void *)) put_device, &chip->dev);
+
         return chip;
 }
 EXPORT_SYMBOL_GPL(tpmm_chip_alloc);
@@ -171,7 +173,7 @@ static int tpm_add_char_device(struct tpm_chip *chip)
 static void tpm_del_char_device(struct tpm_chip *chip)
 {
         cdev_del(&chip->cdev);
-        device_unregister(&chip->dev);
+        device_del(&chip->dev);
 }
 
 static int tpm1_chip_register(struct tpm_chip *chip)