}
 
        /* nRetryCount is 3 in PD2.0 spec where 2 in PD3.0 spec */
-       reg = ((negotiated_rev > PD_REV20 ? PD_RETRY_COUNT_3_0_OR_HIGHER : PD_RETRY_COUNT_DEFAULT)
-              << TCPC_TRANSMIT_RETRY_SHIFT) | (type << TCPC_TRANSMIT_TYPE_SHIFT);
+       reg = FIELD_PREP(TCPC_TRANSMIT_RETRY,
+                        (negotiated_rev > PD_REV20
+                         ? PD_RETRY_COUNT_3_0_OR_HIGHER
+                         : PD_RETRY_COUNT_DEFAULT));
+       reg |= FIELD_PREP(TCPC_TRANSMIT_TYPE, type);
        ret = regmap_write(tcpci->regmap, TCPC_TRANSMIT, reg);
        if (ret < 0)
                return ret;
 
 #define TCPC_RX_DATA                   0x34 /* through 0x4f */
 
 #define TCPC_TRANSMIT                  0x50
-#define TCPC_TRANSMIT_RETRY_SHIFT      4
-#define TCPC_TRANSMIT_RETRY_MASK       0x3
-#define TCPC_TRANSMIT_TYPE_SHIFT       0
-#define TCPC_TRANSMIT_TYPE_MASK                0x7
+#define TCPC_TRANSMIT_RETRY            GENMASK(5, 4)
+#define TCPC_TRANSMIT_TYPE             GENMASK(2, 0)
 
 #define TCPC_TX_BYTE_CNT               0x51
 #define TCPC_TX_HDR                    0x52