return container_of(nb, struct s3c2410_wdt, freq_transition);
 }
 
-static int s3c2410wdt_mask_and_disable_reset(struct s3c2410_wdt *wdt, bool mask)
+static int s3c2410wdt_disable_wdt_reset(struct s3c2410_wdt *wdt, bool mask)
 {
+       const u32 mask_val = BIT(wdt->drv_data->mask_bit);
+       const u32 val = mask ? mask_val : 0;
        int ret;
-       u32 mask_val = 1 << wdt->drv_data->mask_bit;
-       u32 val = 0;
 
-       /* No need to do anything if no PMU CONFIG needed */
-       if (!(wdt->drv_data->quirks & QUIRK_HAS_PMU_CONFIG))
-               return 0;
+       ret = regmap_update_bits(wdt->pmureg, wdt->drv_data->disable_reg,
+                                mask_val, val);
+       if (ret < 0)
+               dev_err(wdt->dev, "failed to update reg(%d)\n", ret);
 
-       if (mask)
-               val = mask_val;
+       return ret;
+}
 
-       if (wdt->drv_data->quirks & QUIRK_HAS_PMU_AUTO_DISABLE) {
-               ret = regmap_update_bits(wdt->pmureg,
-                                        wdt->drv_data->disable_reg, mask_val,
-                                        val);
-               if (ret < 0)
-                       goto error;
-       }
+static int s3c2410wdt_mask_wdt_reset(struct s3c2410_wdt *wdt, bool mask)
+{
+       const u32 mask_val = BIT(wdt->drv_data->mask_bit);
+       const u32 val = mask ? mask_val : 0;
+       int ret;
 
-       ret = regmap_update_bits(wdt->pmureg,
-                       wdt->drv_data->mask_reset_reg,
-                       mask_val, val);
- error:
+       ret = regmap_update_bits(wdt->pmureg, wdt->drv_data->mask_reset_reg,
+                                mask_val, val);
        if (ret < 0)
                dev_err(wdt->dev, "failed to update reg(%d)\n", ret);
 
        return ret;
 }
 
+static int s3c2410wdt_mask_and_disable_reset(struct s3c2410_wdt *wdt, bool mask)
+{
+       int ret;
+
+       if (wdt->drv_data->quirks & QUIRK_HAS_PMU_AUTO_DISABLE) {
+               ret = s3c2410wdt_disable_wdt_reset(wdt, mask);
+               if (ret < 0)
+                       return ret;
+       }
+
+       if (wdt->drv_data->quirks & QUIRK_HAS_PMU_CONFIG) {
+               ret = s3c2410wdt_mask_wdt_reset(wdt, mask);
+               if (ret < 0)
+                       return ret;
+       }
+
+       return 0;
+}
+
 static int s3c2410wdt_keepalive(struct watchdog_device *wdd)
 {
        struct s3c2410_wdt *wdt = watchdog_get_drvdata(wdd);