diff --git a/src/aarch64/cpu-aarch64.cc b/src/aarch64/cpu-aarch64.cc
index 2d2590f4..0438d085 100644
--- a/src/aarch64/cpu-aarch64.cc
+++ b/src/aarch64/cpu-aarch64.cc
@@ -279,8 +279,9 @@ CPUFeatures CPU::InferCPUFeaturesFromOS(
        CPUFeatures::kSVEBitPerm,
        CPUFeatures::kSVESHA3,
        CPUFeatures::kSVESM4,
-       CPUFeatures::kFrintToFixedSizedInt,
+       CPUFeatures::kAXFlag,
        // Bits 40-47
+       CPUFeatures::kFrintToFixedSizedInt,
        CPUFeatures::kSVEI8MM,
        CPUFeatures::kSVEF32MM,
        CPUFeatures::kSVEF64MM,
@@ -288,8 +289,8 @@ CPUFeatures CPU::InferCPUFeaturesFromOS(
        CPUFeatures::kI8MM,
        CPUFeatures::kBF16,
        CPUFeatures::kDGH,
-       CPUFeatures::kRNG,
        // Bits 48+
+       CPUFeatures::kRNG,
        CPUFeatures::kBTI};
 
   uint64_t hwcap_low32 = getauxval(AT_HWCAP);