// SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
 /*
  * Copyright (C) 2015-2017 Intel Deutschland GmbH
- * Copyright (C) 2018-2022 Intel Corporation
+ * Copyright (C) 2018-2023 Intel Corporation
  */
 #include <net/cfg80211.h>
 #include <linux/etherdevice.h>
                                      struct iwl_mvm_pasn_sta *sta)
 {
        list_del(&sta->list);
-       iwl_mvm_rm_sta_id(mvm, vif, sta->int_sta.sta_id);
+
+       if (iwl_mvm_has_mld_api(mvm->fw))
+               iwl_mvm_mld_rm_sta_id(mvm, sta->int_sta.sta_id);
+       else
+               iwl_mvm_rm_sta_id(mvm, vif, sta->int_sta.sta_id);
+
        iwl_mvm_dealloc_int_sta(mvm, &sta->int_sta);
        kfree(sta);
 }