&tc->mod_hdr;
 }
 
-static int mlx5e_tc_attach_mod_hdr(struct mlx5e_priv *priv,
-                                  struct mlx5e_tc_flow *flow,
-                                  struct mlx5_flow_attr *attr)
+int mlx5e_tc_attach_mod_hdr(struct mlx5e_priv *priv,
+                           struct mlx5e_tc_flow *flow,
+                           struct mlx5_flow_attr *attr)
 {
        struct mlx5e_mod_hdr_handle *mh;
 
        return 0;
 }
 
-static void mlx5e_tc_detach_mod_hdr(struct mlx5e_priv *priv,
-                                   struct mlx5e_tc_flow *flow,
-                                   struct mlx5_flow_attr *attr)
+void mlx5e_tc_detach_mod_hdr(struct mlx5e_priv *priv,
+                            struct mlx5e_tc_flow *flow,
+                            struct mlx5_flow_attr *attr)
 {
        /* flow wasn't fully initialized */
        if (!attr->mh)
        return err;
 }
 
-int mlx5e_tc_add_flow_mod_hdr(struct mlx5e_priv *priv,
-                             struct mlx5e_tc_flow *flow,
-                             struct mlx5_flow_attr *attr)
-{
-       struct mlx5e_tc_mod_hdr_acts *mod_hdr_acts = &attr->parse_attr->mod_hdr_acts;
-       struct mlx5_modify_hdr *mod_hdr;
-
-       mod_hdr = mlx5_modify_header_alloc(priv->mdev,
-                                          mlx5e_get_flow_namespace(flow),
-                                          mod_hdr_acts->num_actions,
-                                          mod_hdr_acts->actions);
-       if (IS_ERR(mod_hdr))
-               return PTR_ERR(mod_hdr);
-
-       WARN_ON(attr->modify_hdr);
-       attr->modify_hdr = mod_hdr;
-
-       return 0;
-}
-
 static int
 set_encap_dests(struct mlx5e_priv *priv,
                struct mlx5e_tc_flow *flow,
 static int
 post_process_attr(struct mlx5e_tc_flow *flow,
                  struct mlx5_flow_attr *attr,
-                 bool is_post_act_attr,
                  struct netlink_ext_ack *extack)
 {
        struct mlx5_eswitch *esw = flow->priv->mdev->priv.eswitch;
        }
 
        if (attr->action & MLX5_FLOW_CONTEXT_ACTION_MOD_HDR) {
-               if (vf_tun || is_post_act_attr) {
-                       err = mlx5e_tc_add_flow_mod_hdr(flow->priv, flow, attr);
-                       if (err)
-                               goto err_out;
-               } else {
-                       err = mlx5e_tc_attach_mod_hdr(flow->priv, flow, attr);
-                       if (err)
-                               goto err_out;
-               }
+               err = mlx5e_tc_attach_mod_hdr(flow->priv, flow, attr);
+               if (err)
+                       goto err_out;
        }
 
        if (attr->branch_true &&
            attr->branch_true->action & MLX5_FLOW_CONTEXT_ACTION_MOD_HDR) {
-               err = mlx5e_tc_add_flow_mod_hdr(flow->priv, flow, attr->branch_true);
+               err = mlx5e_tc_attach_mod_hdr(flow->priv, flow, attr->branch_true);
                if (err)
                        goto err_out;
        }
 
        if (attr->branch_false &&
            attr->branch_false->action & MLX5_FLOW_CONTEXT_ACTION_MOD_HDR) {
-               err = mlx5e_tc_add_flow_mod_hdr(flow->priv, flow, attr->branch_false);
+               err = mlx5e_tc_attach_mod_hdr(flow->priv, flow, attr->branch_false);
                if (err)
                        goto err_out;
        }
                esw_attr->int_port = int_port;
        }
 
-       err = post_process_attr(flow, attr, false, extack);
+       err = post_process_attr(flow, attr, extack);
        if (err)
                goto err_out;
 
 
        if (attr->action & MLX5_FLOW_CONTEXT_ACTION_MOD_HDR) {
                mlx5e_mod_hdr_dealloc(&attr->parse_attr->mod_hdr_acts);
-               if (vf_tun && attr->modify_hdr)
-                       mlx5_modify_header_dealloc(priv->mdev, attr->modify_hdr);
-               else
-                       mlx5e_tc_detach_mod_hdr(priv, flow, attr);
+               mlx5e_tc_detach_mod_hdr(priv, flow, attr);
        }
 
        if (attr->action & MLX5_FLOW_CONTEXT_ACTION_COUNT)
                if (err)
                        goto out_free;
 
-               err = post_process_attr(flow, attr, true, extack);
+               err = post_process_attr(flow, attr, extack);
                if (err)
                        goto out_free;
 
 
        if (attr->action & MLX5_FLOW_CONTEXT_ACTION_MOD_HDR) {
                mlx5e_mod_hdr_dealloc(&attr->parse_attr->mod_hdr_acts);
-               if (attr->modify_hdr)
-                       mlx5_modify_header_dealloc(flow->priv->mdev, attr->modify_hdr);
+               mlx5e_tc_detach_mod_hdr(flow->priv, flow, attr);
        }
 }