static bool sun8i_i2s_volatile_reg(struct device *dev, unsigned int reg)
 {
-       if (reg == SUN8I_I2S_INT_STA_REG)
+       switch (reg) {
+       case SUN4I_I2S_FIFO_CTRL_REG:
+       case SUN4I_I2S_FIFO_RX_REG:
+       case SUN4I_I2S_FIFO_STA_REG:
+       case SUN4I_I2S_RX_CNT_REG:
+       case SUN4I_I2S_TX_CNT_REG:
+       case SUN8I_I2S_FIFO_TX_REG:
+       case SUN8I_I2S_INT_STA_REG:
                return true;
-       if (reg == SUN8I_I2S_FIFO_TX_REG)
-               return false;
 
-       return sun4i_i2s_volatile_reg(dev, reg);
+       default:
+               return false;
+       }
 }
 
 static const struct reg_default sun4i_i2s_reg_defaults[] = {