mac_key->high.bits.mac_3 = addr[3];
        mac_key->low.bits.mac_4 = addr[4];
        mac_key->low.bits.mac_5 = addr[5];
-       mac_key->low.bits.vlan = vlan_id;
-       mac_key->low.bits.port = port;
+       dsaf_set_field(mac_key->low.bits.port_vlan, DSAF_TBL_TCAM_KEY_VLAN_M,
+                      DSAF_TBL_TCAM_KEY_VLAN_S, vlan_id);
+       dsaf_set_field(mac_key->low.bits.port_vlan, DSAF_TBL_TCAM_KEY_PORT_M,
+                      DSAF_TBL_TCAM_KEY_PORT_S, port);
+
+       mac_key->low.bits.port_vlan = le16_to_cpu(mac_key->low.bits.port_vlan);
 }
 
 /**
        if (enable) {
                memset(&tbl_tcam_data, 0, sizeof(tbl_tcam_data));
                memset(&tbl_tcam_mask, 0, sizeof(tbl_tcam_mask));
-               tbl_tcam_data.low.bits.port = port;
-               tbl_tcam_mask.low.bits.port = 0xf;  /* [3:0]: port id */
+               dsaf_set_field(tbl_tcam_data.low.bits.port_vlan,
+                              DSAF_TBL_TCAM_KEY_PORT_M,
+                              DSAF_TBL_TCAM_KEY_PORT_S, port);
+               dsaf_set_field(tbl_tcam_mask.low.bits.port_vlan,
+                              DSAF_TBL_TCAM_KEY_PORT_M,
+                              DSAF_TBL_TCAM_KEY_PORT_S, 0xf);
 
                /* SUB_QID */
                dsaf_set_bit(mac_data.tbl_mcast_port_msk[0],
 
        return (void *)((u8 *)dsaf_dev + sizeof(*dsaf_dev));
 }
 
+#define DSAF_TBL_TCAM_KEY_PORT_S 0
+#define DSAF_TBL_TCAM_KEY_PORT_M (((1ULL << 4) - 1) << 0)
+#define DSAF_TBL_TCAM_KEY_VLAN_S 4
+#define DSAF_TBL_TCAM_KEY_VLAN_M (((1ULL << 12) - 1) << 4)
+
 struct dsaf_drv_tbl_tcam_key {
        union {
                struct {
        } high;
        union {
                struct {
-                       u32 port:4; /* port id, */
-                       /* dsaf-mode fixed 0, non-dsaf-mode port id*/
-                       u32 vlan:12; /* vlan id */
-                       u32 mac_5:8;
-                       u32 mac_4:8;
+                       u16 port_vlan;
+                       u8 mac_5;
+                       u8 mac_4;
                } bits;
 
                u32 val;