1 files changed, 3 insertions, 3 deletions

diff --git a/drivers/mtd/ubi/wl.c b/drivers/mtd/ubi/wl.c
index d5076d2cce39..bf13ca275f85 100644
--- a/drivers/mtd/ubi/wl.c
+++ b/drivers/mtd/ubi/wl.c
@@ -1582,7 +1582,7 @@ static int paranoid_check_ec(struct ubi_device *ubi, int pnum, int ec)
         long long read_ec;
         struct ubi_ec_hdr *ec_hdr;
 
-        if (!(ubi_chk_flags & UBI_CHK_GEN))
+        if (!ubi->dbg->chk_gen)
                 return 0;
 
         ec_hdr = kzalloc(ubi->ec_hdr_alsize, GFP_NOFS);
@@ -1623,7 +1623,7 @@ static int paranoid_check_in_wl_tree(const struct ubi_device *ubi,
                                      struct ubi_wl_entry *e,
                                      struct rb_root *root)
 {
-        if (!(ubi_chk_flags & UBI_CHK_GEN))
+        if (!ubi->dbg->chk_gen)
                 return 0;
 
         if (in_wl_tree(e, root))
@@ -1649,7 +1649,7 @@ static int paranoid_check_in_pq(const struct ubi_device *ubi,
         struct ubi_wl_entry *p;
         int i;
 
-        if (!(ubi_chk_flags & UBI_CHK_GEN))
+        if (!ubi->dbg->chk_gen)
                 return 0;
 
         for (i = 0; i < UBI_PROT_QUEUE_LEN; ++i)