1 files changed, 1 insertions, 1 deletions

diff --git a/block/blk-integrity.c b/block/blk-integrity.c
index 69023da63151..e3817a016a12 100644
--- a/block/blk-integrity.c
+++ b/block/blk-integrity.c
@@ -376,7 +376,7 @@ void blk_integrity_unregister(struct gendisk *disk)
 
         kobject_uevent(&bi->kobj, KOBJ_REMOVE);
         kobject_del(&bi->kobj);
-        kobject_put(&disk_to_dev(disk)->kobj);
         kmem_cache_free(integrity_cachep, bi);
+        disk->integrity = NULL;
 }
 EXPORT_SYMBOL(blk_integrity_unregister);