* ice_sched_remove_elems - remove nodes from HW
  * @hw: pointer to the HW struct
  * @parent: pointer to the parent node
- * @num_nodes: number of nodes
- * @node_teids: array of node teids to be deleted
+ * @node_teid: node teid to be deleted
  *
  * This function remove nodes from HW
  */
 static int
 ice_sched_remove_elems(struct ice_hw *hw, struct ice_sched_node *parent,
-                      u16 num_nodes, u32 *node_teids)
+                      u32 node_teid)
 {
-       struct ice_aqc_delete_elem *buf;
-       u16 i, num_groups_removed = 0;
-       u16 buf_size;
+       DEFINE_FLEX(struct ice_aqc_delete_elem, buf, teid, 1);
+       u16 buf_size = __struct_size(buf);
+       u16 num_groups_removed = 0;
        int status;
 
-       buf_size = struct_size(buf, teid, num_nodes);
-       buf = devm_kzalloc(ice_hw_to_dev(hw), buf_size, GFP_KERNEL);
-       if (!buf)
-               return -ENOMEM;
-
        buf->hdr.parent_teid = parent->info.node_teid;
-       buf->hdr.num_elems = cpu_to_le16(num_nodes);
-       for (i = 0; i < num_nodes; i++)
-               buf->teid[i] = cpu_to_le32(node_teids[i]);
+       buf->hdr.num_elems = cpu_to_le16(1);
+       buf->teid[0] = cpu_to_le32(node_teid);
 
        status = ice_aq_delete_sched_elems(hw, 1, buf, buf_size,
                                           &num_groups_removed, NULL);
                ice_debug(hw, ICE_DBG_SCHED, "remove node failed FW error %d\n",
                          hw->adminq.sq_last_status);
 
-       devm_kfree(ice_hw_to_dev(hw), buf);
        return status;
 }
 
            node->info.data.elem_type != ICE_AQC_ELEM_TYPE_LEAF) {
                u32 teid = le32_to_cpu(node->info.node_teid);
 
-               ice_sched_remove_elems(hw, node->parent, 1, &teid);
+               ice_sched_remove_elems(hw, node->parent, teid);
        }
        parent = node->parent;
        /* root has no parent */
                int status;
 
                /* remove the default leaf node */
-               status = ice_sched_remove_elems(pi->hw, node->parent, 1, &teid);
+               status = ice_sched_remove_elems(pi->hw, node->parent, teid);
                if (!status)
                        ice_free_sched_node(pi, node);
        }