1 files changed, 5 insertions, 9 deletions

diff --git a/drivers/lightnvm/pblk-init.c b/drivers/lightnvm/pblk-init.c
index 0e37104de596..72ad3e70318c 100644
--- a/drivers/lightnvm/pblk-init.c
+++ b/drivers/lightnvm/pblk-init.c
@@ -347,23 +347,19 @@ fail_destroy_ws:
 
 static int pblk_get_global_caches(void)
 {
-        int ret;
+        int ret = 0;
 
         mutex_lock(&pblk_caches.mutex);
 
-        if (kref_read(&pblk_caches.kref) > 0) {
-                kref_get(&pblk_caches.kref);
-                mutex_unlock(&pblk_caches.mutex);
-                return 0;
-        }
+        if (kref_get_unless_zero(&pblk_caches.kref))
+                goto out;
 
         ret = pblk_create_global_caches();
-
         if (!ret)
-                kref_get(&pblk_caches.kref);
+                kref_init(&pblk_caches.kref);
 
+out:
         mutex_unlock(&pblk_caches.mutex);
-
         return ret;
 }