1 files changed, 2 insertions, 1 deletions

diff --git a/drivers/lightnvm/pblk-init.c b/drivers/lightnvm/pblk-init.c
index b2c49fc006c9..e0db0cb3122d 100644
--- a/drivers/lightnvm/pblk-init.c
+++ b/drivers/lightnvm/pblk-init.c
@@ -195,7 +195,8 @@ static int pblk_rwb_init(struct pblk *pblk)
         unsigned long buffer_size;
         int pgs_in_buffer;
 
-        pgs_in_buffer = max(geo->mw_cunits, geo->ws_opt) * geo->all_luns;
+        pgs_in_buffer = (max(geo->mw_cunits, geo->ws_opt) + geo->ws_opt)
+                                                                * geo->all_luns;
 
         if (write_buffer_size && (write_buffer_size > pgs_in_buffer))
                 buffer_size = write_buffer_size;