return ret;
 }
 
+/**
+ * i40e_vsi_reconfig_tc - Reconfigure VSI Tx Scheduler for stored TC map
+ * @vsi: VSI to be reconfigured
+ *
+ * This reconfigures a particular VSI for TCs that are mapped to the
+ * TC bitmap stored previously for the VSI.
+ *
+ * Context: It is expected that the VSI queues have been quisced before
+ *          calling this function.
+ *
+ * Return: 0 on success, negative value on failure
+ **/
+static int i40e_vsi_reconfig_tc(struct i40e_vsi *vsi)
+{
+       u8 enabled_tc;
+
+       enabled_tc = vsi->tc_config.enabled_tc;
+       vsi->tc_config.enabled_tc = 0;
+
+       return i40e_vsi_config_tc(vsi, enabled_tc);
+}
+
 /**
  * i40e_get_link_speed - Returns link speed for the interface
  * @vsi: VSI to be configured
        struct i40e_vsi *main_vsi;
        u16 alloc_queue_pairs;
        struct i40e_pf *pf;
-       u8 enabled_tc;
        int ret;
 
        if (!vsi)
         * layout configurations.
         */
        main_vsi = i40e_pf_get_main_vsi(pf);
-       enabled_tc = main_vsi->tc_config.enabled_tc;
-       main_vsi->tc_config.enabled_tc = 0;
        main_vsi->seid = pf->main_vsi_seid;
-       i40e_vsi_config_tc(main_vsi, enabled_tc);
+       i40e_vsi_reconfig_tc(main_vsi);
 
        if (vsi->type == I40E_VSI_MAIN)
                i40e_rm_default_mac_filter(vsi, pf->hw.mac.perm_addr);
                }
        } else {
                /* force a reset of TC and queue layout configurations */
-               u8 enabled_tc = main_vsi->tc_config.enabled_tc;
-
-               main_vsi->tc_config.enabled_tc = 0;
                main_vsi->seid = pf->main_vsi_seid;
-               i40e_vsi_config_tc(main_vsi, enabled_tc);
+               i40e_vsi_reconfig_tc(main_vsi);
        }
        i40e_vlan_stripping_disable(main_vsi);