1 files changed, 4 insertions, 1 deletions

diff --git a/drivers/block/umem.c b/drivers/block/umem.c
index dff3766f117f..5872036e8ae6 100644
--- a/drivers/block/umem.c
+++ b/drivers/block/umem.c
@@ -1179,8 +1179,10 @@ static int __init mm_init(void)
                 return -ENOMEM;
 
         err = major_nr = register_blkdev(0, "umem");
-        if (err < 0)
+        if (err < 0) {
+                pci_unregister_driver(&mm_pci_driver);
                 return -EIO;
+        }
 
         for (i = 0; i < num_cards; i++) {
                 mm_gendisk[i] = alloc_disk(1 << MM_SHIFT);
@@ -1207,6 +1209,7 @@ static int __init mm_init(void)
         return 0;
 
 out:
+        pci_unregister_driver(&mm_pci_driver);
         unregister_blkdev(major_nr, "umem");
         while (i--)
                 put_disk(mm_gendisk[i]);