1 files changed, 4 insertions, 0 deletions

diff --git a/drivers/bcma/main.c b/drivers/bcma/main.c
index 2c1798e38abd..38688236b3cd 100644
--- a/drivers/bcma/main.c
+++ b/drivers/bcma/main.c
@@ -633,8 +633,11 @@ static int bcma_device_probe(struct device *dev)
                                                drv);
         int err = 0;
 
+        get_device(dev);
         if (adrv->probe)
                 err = adrv->probe(core);
+        if (err)
+                put_device(dev);
 
         return err;
 }
@@ -647,6 +650,7 @@ static int bcma_device_remove(struct device *dev)
 
         if (adrv->remove)
                 adrv->remove(core);
+        put_device(dev);
 
         return 0;
 }