#define AWCC_THERMAL_MODE_GMODE                        0xAB
 
 #define AWCC_FAILURE_CODE                      0xFFFFFFFF
-#define AWCC_THERMAL_TABLE_MASK                        GENMASK(7, 4)
+
+#define AWCC_SENSOR_ID_FLAG                    BIT(8)
 #define AWCC_THERMAL_MODE_MASK                 GENMASK(3, 0)
-#define AWCC_SENSOR_ID_MASK                    BIT(8)
+#define AWCC_THERMAL_TABLE_MASK                        GENMASK(7, 4)
+#define AWCC_RESOURCE_ID_MASK                  GENMASK(7, 0)
 
 static bool force_platform_profile;
 module_param_unsafe(force_platform_profile, bool, 0);
 };
 
 enum AWCC_THERMAL_TABLES {
-       AWCC_THERMAL_TABLE_LEGACY               = 0x90,
-       AWCC_THERMAL_TABLE_USTT                 = 0xA0,
+       AWCC_THERMAL_TABLE_LEGACY               = 0x9,
+       AWCC_THERMAL_TABLE_USTT                 = 0xA,
 };
 
 enum awcc_thermal_profile {
  * Thermal Profile control
  *  - Provides thermal profile control through the Platform Profile API
  */
-static bool is_awcc_thermal_mode(u32 code)
+static bool is_awcc_thermal_profile_id(u8 code)
 {
-       if (code & AWCC_SENSOR_ID_MASK)
-               return false;
+       u8 table = FIELD_GET(AWCC_THERMAL_TABLE_MASK, code);
+       u8 mode = FIELD_GET(AWCC_THERMAL_MODE_MASK, code);
 
-       if ((code & AWCC_THERMAL_MODE_MASK) >= AWCC_PROFILE_LAST)
+       if (mode >= AWCC_PROFILE_LAST)
                return false;
 
-       if ((code & AWCC_THERMAL_TABLE_MASK) == AWCC_THERMAL_TABLE_LEGACY &&
-           (code & AWCC_THERMAL_MODE_MASK) >= AWCC_PROFILE_LEGACY_QUIET)
+       if (table == AWCC_THERMAL_TABLE_LEGACY && mode >= AWCC_PROFILE_LEGACY_QUIET)
                return true;
 
-       if ((code & AWCC_THERMAL_TABLE_MASK) == AWCC_THERMAL_TABLE_USTT &&
-           (code & AWCC_THERMAL_MODE_MASK) <= AWCC_PROFILE_USTT_LOW_POWER)
+       if (table == AWCC_THERMAL_TABLE_USTT && mode <= AWCC_PROFILE_USTT_LOW_POWER)
                return true;
 
        return false;
                return 0;
        }
 
-       if (!is_awcc_thermal_mode(out_data))
+       if (!is_awcc_thermal_profile_id(out_data))
                return -ENODATA;
 
-       out_data &= AWCC_THERMAL_MODE_MASK;
+       out_data = FIELD_GET(AWCC_THERMAL_MODE_MASK, out_data);
        *profile = awcc_mode_to_platform_profile[out_data];
 
        return 0;
        u32 first_mode;
        u32 out_data;
        int ret;
+       u8 id;
 
        ret = awcc_thermal_information(priv->wdev, AWCC_OP_GET_SYSTEM_DESCRIPTION,
                                       0, (u32 *) &sys_desc);
                if (ret == -EBADRQC)
                        break;
 
-               if (!is_awcc_thermal_mode(out_data))
+               id = FIELD_GET(AWCC_RESOURCE_ID_MASK, out_data);
+               if (!is_awcc_thermal_profile_id(id))
                        continue;
 
-               mode = out_data & AWCC_THERMAL_MODE_MASK;
+               mode = FIELD_GET(AWCC_THERMAL_MODE_MASK, id);
                profile = awcc_mode_to_platform_profile[mode];
-               priv->supported_thermal_profiles[profile] = out_data;
+               priv->supported_thermal_profiles[profile] = id;
 
                set_bit(profile, choices);
        }