1 files changed, 2 insertions, 0 deletions

diff --git a/drivers/mtd/devices/block2mtd.c b/drivers/mtd/devices/block2mtd.c
index 8debac9efde9..e081bfeaaf7d 100644
--- a/drivers/mtd/devices/block2mtd.c
+++ b/drivers/mtd/devices/block2mtd.c
@@ -62,6 +62,7 @@ static int _block2mtd_erase(struct block2mtd_dev *dev, loff_t to, size_t len)
                                 memset(page_address(page), 0xff, PAGE_SIZE);
                                 set_page_dirty(page);
                                 unlock_page(page);
+                                balance_dirty_pages_ratelimited(mapping);
                                 break;
                         }
 
@@ -152,6 +153,7 @@ static int _block2mtd_write(struct block2mtd_dev *dev, const u_char *buf,
                         memcpy(page_address(page) + offset, buf, cpylen);
                         set_page_dirty(page);
                         unlock_page(page);
+                        balance_dirty_pages_ratelimited(mapping);
                 }
                 page_cache_release(page);