#include <linux/regmap.h>
 
 #define MTK_PMIC_RST_DU_MASK   GENMASK(9, 8)
-#define MTK_PMIC_RST_DU_SHIFT  8
 #define MTK_PMIC_PWRKEY_RST    BIT(6)
 #define MTK_PMIC_HOMEKEY_RST   BIT(5)
 
 struct mtk_pmic_regs {
        const struct mtk_pmic_keys_regs keys_regs[MTK_PMIC_MAX_KEY_COUNT];
        u32 pmic_rst_reg;
+       u32 rst_lprst_mask; /* Long-press reset timeout bitmask */
 };
 
 static const struct mtk_pmic_regs mt6397_regs = {
                MTK_PMIC_KEYS_REGS(MT6397_OCSTATUS2,
                0x10, MT6397_INT_RSV, 0x8, MTK_PMIC_HOMEKEY_RST),
        .pmic_rst_reg = MT6397_TOP_RST_MISC,
+       .rst_lprst_mask = MTK_PMIC_RST_DU_MASK,
 };
 
 static const struct mtk_pmic_regs mt6323_regs = {
                MTK_PMIC_KEYS_REGS(MT6323_CHRSTATUS,
                0x4, MT6323_INT_MISC_CON, 0x8, MTK_PMIC_HOMEKEY_RST),
        .pmic_rst_reg = MT6323_TOP_RST_MISC,
+       .rst_lprst_mask = MTK_PMIC_RST_DU_MASK,
 };
 
 static const struct mtk_pmic_regs mt6358_regs = {
                                   0x8, MT6358_PSC_TOP_INT_CON0, 0xa,
                                   MTK_PMIC_HOMEKEY_RST),
        .pmic_rst_reg = MT6358_TOP_RST_MISC,
+       .rst_lprst_mask = MTK_PMIC_RST_DU_MASK,
 };
 
 struct mtk_pmic_keys_info {
 };
 
 static void mtk_pmic_keys_lp_reset_setup(struct mtk_pmic_keys *keys,
-                                        u32 pmic_rst_reg)
+                                        const struct mtk_pmic_regs *regs)
 {
        const struct mtk_pmic_keys_regs *kregs_home, *kregs_pwr;
        u32 long_press_mode, long_press_debounce;
        if (error)
                long_press_debounce = 0;
 
-       mask = MTK_PMIC_RST_DU_MASK;
-       value = long_press_debounce << MTK_PMIC_RST_DU_SHIFT;
+       mask = regs->rst_lprst_mask;
+       value = long_press_debounce << (ffs(regs->rst_lprst_mask) - 1);
 
        error  = of_property_read_u32(keys->dev->of_node,
                                      "mediatek,long-press-mode",
                break;
        }
 
-       regmap_update_bits(keys->regmap, pmic_rst_reg, mask, value);
+       regmap_update_bits(keys->regmap, regs->pmic_rst_reg, mask, value);
 }
 
 static irqreturn_t mtk_pmic_keys_irq_handler_thread(int irq, void *data)
                return error;
        }
 
-       mtk_pmic_keys_lp_reset_setup(keys, mtk_pmic_regs->pmic_rst_reg);
+       mtk_pmic_keys_lp_reset_setup(keys, mtk_pmic_regs);
 
        platform_set_drvdata(pdev, keys);