2 files changed, 4 insertions, 4 deletions

diff --git a/drivers/mtd/ubi/fastmap.c b/drivers/mtd/ubi/fastmap.c
index 154275182b4b..f5aa4b02cfa6 100644
--- a/drivers/mtd/ubi/fastmap.c
+++ b/drivers/mtd/ubi/fastmap.c
@@ -1343,7 +1343,7 @@ out:
 static int invalidate_fastmap(struct ubi_device *ubi,
                               struct ubi_fastmap_layout *fm)
 {
-        int ret, i;
+        int ret;
         struct ubi_vid_hdr *vh;
 
         ret = erase_block(ubi, fm->e[0]->pnum);
@@ -1360,9 +1360,6 @@ static int invalidate_fastmap(struct ubi_device *ubi,
         vh->sqnum = cpu_to_be64(ubi_next_sqnum(ubi));
         ret = ubi_io_write_vid_hdr(ubi, fm->e[0]->pnum, vh);
 
-        for (i = 0; i < fm->used_blocks; i++)
-                ubi_wl_put_fm_peb(ubi, fm->e[i], i, fm->to_be_tortured[i]);
-
         return ret;
 }
 
diff --git a/drivers/mtd/ubi/wl.c b/drivers/mtd/ubi/wl.c
index 5df49d3cb5c7..c95bfb183c62 100644
--- a/drivers/mtd/ubi/wl.c
+++ b/drivers/mtd/ubi/wl.c
@@ -1069,6 +1069,9 @@ static int wear_leveling_worker(struct ubi_device *ubi, struct ubi_work *wrk,
                 if (!(e2->ec - e1->ec >= UBI_WL_THRESHOLD)) {
                         dbg_wl("no WL needed: min used EC %d, max free EC %d",
                                e1->ec, e2->ec);
+
+                        /* Give the unused PEB back */
+                        wl_tree_add(e2, &ubi->free);
                         goto out_cancel;
                 }
                 self_check_in_wl_tree(ubi, e1, &ubi->used);