#define ARM64_WORKAROUND_1165522               37
 #define ARM64_HAS_ADDRESS_AUTH_ARCH            38
 #define ARM64_HAS_ADDRESS_AUTH_IMP_DEF         39
-#define ARM64_HAS_ADDRESS_AUTH                 40
-#define ARM64_HAS_GENERIC_AUTH_ARCH            41
-#define ARM64_HAS_GENERIC_AUTH_IMP_DEF         42
-#define ARM64_HAS_GENERIC_AUTH                 43
+#define ARM64_HAS_GENERIC_AUTH_ARCH            40
+#define ARM64_HAS_GENERIC_AUTH_IMP_DEF         41
 
-#define ARM64_NCAPS                            44
+#define ARM64_NCAPS                            42
 
 #endif /* __ASM_CPUCAPS_H */
 
 static inline bool system_supports_address_auth(void)
 {
        return IS_ENABLED(CONFIG_ARM64_PTR_AUTH) &&
-               cpus_have_const_cap(ARM64_HAS_ADDRESS_AUTH);
+               (cpus_have_const_cap(ARM64_HAS_ADDRESS_AUTH_ARCH) ||
+                cpus_have_const_cap(ARM64_HAS_ADDRESS_AUTH_IMP_DEF));
 }
 
 static inline bool system_supports_generic_auth(void)
 {
        return IS_ENABLED(CONFIG_ARM64_PTR_AUTH) &&
-               cpus_have_const_cap(ARM64_HAS_GENERIC_AUTH);
+               (cpus_have_const_cap(ARM64_HAS_GENERIC_AUTH_ARCH) ||
+                cpus_have_const_cap(ARM64_HAS_GENERIC_AUTH_IMP_DEF));
 }
 
 #define ARM64_SSBD_UNKNOWN             -1
 
                .field_pos = ID_AA64ISAR1_APA_SHIFT,
                .min_field_value = ID_AA64ISAR1_APA_ARCHITECTED,
                .matches = has_cpuid_feature,
+               .cpu_enable = cpu_enable_address_auth,
        },
        {
                .desc = "Address authentication (IMP DEF algorithm)",
                .field_pos = ID_AA64ISAR1_API_SHIFT,
                .min_field_value = ID_AA64ISAR1_API_IMP_DEF,
                .matches = has_cpuid_feature,
-       },
-       {
-               .capability = ARM64_HAS_ADDRESS_AUTH,
-               .type = ARM64_CPUCAP_SYSTEM_FEATURE,
-               .matches = has_address_auth,
                .cpu_enable = cpu_enable_address_auth,
        },
        {
                .min_field_value = ID_AA64ISAR1_GPI_IMP_DEF,
                .matches = has_cpuid_feature,
        },
-       {
-               .capability = ARM64_HAS_GENERIC_AUTH,
-               .type = ARM64_CPUCAP_SYSTEM_FEATURE,
-               .matches = has_generic_auth,
-       },
 #endif /* CONFIG_ARM64_PTR_AUTH */
        {},
 };