static const struct field_mapping fields[] = {
        TD_SYSINFO_MAP(MAX_TDMRS,             max_tdmrs),
        TD_SYSINFO_MAP(MAX_RESERVED_PER_TDMR, max_reserved_per_tdmr),
-       TD_SYSINFO_MAP(PAMT_4K_ENTRY_SIZE,    pamt_entry_size[TDX_PS_4K]),
-       TD_SYSINFO_MAP(PAMT_2M_ENTRY_SIZE,    pamt_entry_size[TDX_PS_2M]),
-       TD_SYSINFO_MAP(PAMT_1G_ENTRY_SIZE,    pamt_entry_size[TDX_PS_1G]),
+       TD_SYSINFO_MAP(PAMT_4K_ENTRY_SIZE,    pamt_4k_entry_size),
+       TD_SYSINFO_MAP(PAMT_2M_ENTRY_SIZE,    pamt_2m_entry_size),
+       TD_SYSINFO_MAP(PAMT_1G_ENTRY_SIZE,    pamt_1g_entry_size),
 };
 
 static int get_tdx_sys_info_tdmr(struct tdx_sys_info_tdmr *sysinfo_tdmr)
                           struct tdmr_info_list *tdmr_list,
                           struct tdx_sys_info_tdmr *sysinfo_tdmr)
 {
+       u16 pamt_entry_size[TDX_PS_NR] = {
+               sysinfo_tdmr->pamt_4k_entry_size,
+               sysinfo_tdmr->pamt_2m_entry_size,
+               sysinfo_tdmr->pamt_1g_entry_size,
+       };
        int ret;
 
        ret = fill_out_tdmrs(tmb_list, tdmr_list);
        if (ret)
                return ret;
 
-       ret = tdmrs_set_up_pamt_all(tdmr_list, tmb_list,
-                       sysinfo_tdmr->pamt_entry_size);
+       ret = tdmrs_set_up_pamt_all(tdmr_list, tmb_list, pamt_entry_size);
        if (ret)
                return ret;
 
 
 struct tdx_sys_info_tdmr {
        u16 max_tdmrs;
        u16 max_reserved_per_tdmr;
-       u16 pamt_entry_size[TDX_PS_NR];
+       u16 pamt_4k_entry_size;
+       u16 pamt_2m_entry_size;
+       u16 pamt_1g_entry_size;
 };
 
 /* Kernel used global metadata fields */