}
 EXPORT_SYMBOL_IF_IWLWIFI_KUNIT(iwl_mvm_get_link_grade);
 
+static
 u8 iwl_mvm_set_link_selection_data(struct ieee80211_vif *vif,
                                   struct iwl_mvm_link_sel_data *data,
                                   unsigned long usable_links,
 
        return !(mvmvif->esr_disable_reason & ~IWL_MVM_ESR_BLOCKED_COEX);
 }
 
-/*
- * This function receives a bitmap of usable links and check if we can enter
- * eSR on those links.
- */
-static bool iwl_mvm_can_enter_esr(struct iwl_mvm *mvm,
-                                 struct ieee80211_vif *vif,
-                                 unsigned long desired_links)
-{
-       struct iwl_mvm_link_sel_data data[IEEE80211_MLD_MAX_NUM_LINKS];
-       u8 best_link, n_data;
-
-       if (!iwl_mvm_esr_allowed_on_vif(mvm, vif))
-               return false;
-
-       n_data = iwl_mvm_set_link_selection_data(vif, data, desired_links,
-                                                &best_link);
-
-       if (n_data != 2)
-               return false;
-
-
-       return iwl_mvm_mld_valid_link_pair(vif, &data[0], &data[1]);
-}
-
 static bool iwl_mvm_mld_can_activate_links(struct ieee80211_hw *hw,
                                           struct ieee80211_vif *vif,
                                           u16 desired_links)
        }
 
        /* If it is an eSR device, check that we can enter eSR */
-       if (iwl_mvm_is_esr_supported(mvm->fwrt.trans))
-               ret = iwl_mvm_can_enter_esr(mvm, vif, desired_links);
+       ret = iwl_mvm_is_esr_supported(mvm->fwrt.trans) &&
+             iwl_mvm_esr_allowed_on_vif(mvm, vif);
+
 unlock:
        mutex_unlock(&mvm->mutex);
        return ret;
 
 u8 iwl_mvm_get_primary_link(struct ieee80211_vif *vif);
 u8 iwl_mvm_get_other_link(struct ieee80211_vif *vif, u8 link_id);
 
-#if IS_ENABLED(CONFIG_IWLWIFI_KUNIT_TESTS)
-unsigned int iwl_mvm_get_link_grade(struct ieee80211_bss_conf *link_conf);
-#endif
-
 struct iwl_mvm_link_sel_data {
        u8 link_id;
        enum nl80211_band band;
        u16 grade;
 };
 
-u8 iwl_mvm_set_link_selection_data(struct ieee80211_vif *vif,
-                                  struct iwl_mvm_link_sel_data *data,
-                                  unsigned long usable_links,
-                                  u8 *best_link_idx);
+#if IS_ENABLED(CONFIG_IWLWIFI_KUNIT_TESTS)
+unsigned int iwl_mvm_get_link_grade(struct ieee80211_bss_conf *link_conf);
 bool iwl_mvm_mld_valid_link_pair(struct ieee80211_vif *vif,
                                 const struct iwl_mvm_link_sel_data *a,
                                 const struct iwl_mvm_link_sel_data *b);
+#endif
 
 /* AP and IBSS */
 bool iwl_mvm_start_ap_ibss_common(struct ieee80211_hw *hw,