*/
 static enum ice_ddp_state ice_get_pkg_info(struct ice_hw *hw)
 {
-       enum ice_ddp_state state = ICE_DDP_PKG_SUCCESS;
-       struct ice_aqc_get_pkg_info_resp *pkg_info;
-       u16 size;
+       DEFINE_FLEX(struct ice_aqc_get_pkg_info_resp, pkg_info, pkg_info,
+                   ICE_PKG_CNT);
+       u16 size = __struct_size(pkg_info);
        u32 i;
 
-       size = struct_size(pkg_info, pkg_info, ICE_PKG_CNT);
-       pkg_info = kzalloc(size, GFP_KERNEL);
-       if (!pkg_info)
+       if (ice_aq_get_pkg_info_list(hw, pkg_info, size, NULL))
                return ICE_DDP_PKG_ERR;
 
-       if (ice_aq_get_pkg_info_list(hw, pkg_info, size, NULL)) {
-               state = ICE_DDP_PKG_ERR;
-               goto init_pkg_free_alloc;
-       }
-
        for (i = 0; i < le32_to_cpu(pkg_info->count); i++) {
 #define ICE_PKG_FLAG_COUNT 4
                char flags[ICE_PKG_FLAG_COUNT + 1] = { 0 };
                          pkg_info->pkg_info[i].name, flags);
        }
 
-init_pkg_free_alloc:
-       kfree(pkg_info);
-
-       return state;
+       return ICE_DDP_PKG_SUCCESS;
 }
 
 /**
                                             struct ice_pkg_hdr *ospkg,
                                             struct ice_seg **seg)
 {
-       struct ice_aqc_get_pkg_info_resp *pkg;
+       DEFINE_FLEX(struct ice_aqc_get_pkg_info_resp, pkg, pkg_info,
+                   ICE_PKG_CNT);
+       u16 size = __struct_size(pkg);
        enum ice_ddp_state state;
-       u16 size;
        u32 i;
 
        /* Check package version compatibility */
        }
 
        /* Check if FW is compatible with the OS package */
-       size = struct_size(pkg, pkg_info, ICE_PKG_CNT);
-       pkg = kzalloc(size, GFP_KERNEL);
-       if (!pkg)
-               return ICE_DDP_PKG_ERR;
-
-       if (ice_aq_get_pkg_info_list(hw, pkg, size, NULL)) {
-               state = ICE_DDP_PKG_LOAD_ERROR;
-               goto fw_ddp_compat_free_alloc;
-       }
+       if (ice_aq_get_pkg_info_list(hw, pkg, size, NULL))
+               return ICE_DDP_PKG_LOAD_ERROR;
 
        for (i = 0; i < le32_to_cpu(pkg->count); i++) {
                /* loop till we find the NVM package */
                /* done processing NVM package so break */
                break;
        }
-fw_ddp_compat_free_alloc:
-       kfree(pkg);
+
        return state;
 }