struct spi_mem_op op;
        u8 *buf = nor->bouncebuf;
        int ret;
+       u8 addr_mode_nbytes = nor->params->addr_mode_nbytes;
 
        /* Use 20 dummy cycles for memory array reads. */
        *buf = 20;
        op = (struct spi_mem_op)
-               MICRON_ST_NOR_WR_ANY_REG_OP(3, SPINOR_REG_MT_CFR1V, 1, buf);
+               MICRON_ST_NOR_WR_ANY_REG_OP(addr_mode_nbytes,
+                                           SPINOR_REG_MT_CFR1V, 1, buf);
        ret = spi_nor_write_any_volatile_reg(nor, &op, nor->reg_proto);
        if (ret)
                return ret;
 
        buf[0] = SPINOR_MT_OCT_DTR;
        op = (struct spi_mem_op)
-               MICRON_ST_NOR_WR_ANY_REG_OP(3, SPINOR_REG_MT_CFR0V, 1, buf);
+               MICRON_ST_NOR_WR_ANY_REG_OP(addr_mode_nbytes,
+                                           SPINOR_REG_MT_CFR0V, 1, buf);
        ret = spi_nor_write_any_volatile_reg(nor, &op, nor->reg_proto);
        if (ret)
                return ret;
        buf[0] = SPINOR_MT_EXSPI;
        buf[1] = SPINOR_REG_MT_CFR1V_DEF;
        op = (struct spi_mem_op)
-               MICRON_ST_NOR_WR_ANY_REG_OP(4, SPINOR_REG_MT_CFR0V, 2, buf);
+               MICRON_ST_NOR_WR_ANY_REG_OP(nor->addr_nbytes,
+                                           SPINOR_REG_MT_CFR0V, 2, buf);
        ret = spi_nor_write_any_volatile_reg(nor, &op, SNOR_PROTO_8_8_8_DTR);
        if (ret)
                return ret;