diff --git a/arith/benches/ext_field.rs b/arith/benches/ext_field.rs
index 41bf922e..3c3d82d8 100644
--- a/arith/benches/ext_field.rs
+++ b/arith/benches/ext_field.rs
@@ -1,4 +1,4 @@
-use arith::{ExtensionField, Field, GF2_128x8, M31Ext3, M31Ext3x16, GF2_128, GF2_128x8_256};
+use arith::{ExtensionField, Field, GF2_128x8, GF2_128x8_256, M31Ext3, M31Ext3x16, GF2_128};
 use ark_std::test_rng;
 use criterion::{criterion_group, criterion_main, BatchSize, Criterion};
 use tynm::type_name;
diff --git a/arith/benches/field.rs b/arith/benches/field.rs
index b276edc2..22eab434 100644
--- a/arith/benches/field.rs
+++ b/arith/benches/field.rs
@@ -1,6 +1,9 @@
 // this module benchmarks the performance of different field operations
 
-use arith::{Field, GF2_128x8, GF2x8, M31Ext3, M31Ext3x16, M31x16, GF2, GF2_128, M31, M31x16_256, GF2_128x8_256};
+use arith::{
+    Field, GF2_128x8, GF2_128x8_256, GF2x8, M31Ext3, M31Ext3x16, M31x16, M31x16_256, GF2, GF2_128,
+    M31,
+};
 use ark_std::test_rng;
 use criterion::{criterion_group, criterion_main, BatchSize, Criterion};
 use halo2curves::bn256::Fr;
diff --git a/arith/src/extension_field/gf2_128x8.rs b/arith/src/extension_field/gf2_128x8.rs
index cc9d212f..ba393103 100644
--- a/arith/src/extension_field/gf2_128x8.rs
+++ b/arith/src/extension_field/gf2_128x8.rs
@@ -3,13 +3,13 @@ pub(crate) mod neon;
 #[cfg(target_arch = "aarch64")]
 pub type GF2_128x8 = neon::NeonGF2_128x8;
 
-#[cfg(target_arch = "x86_64")]
-mod avx256;
 #[cfg(target_arch = "x86_64")]
 mod avx;
 #[cfg(target_arch = "x86_64")]
+mod avx256;
+#[cfg(target_arch = "x86_64")]
 pub type GF2_128x8_256 = avx256::AVX256GF2_128x8;
-#[cfg(feature = "avx256")]
+#[cfg(all(target_arch = "x86_64", feature = "avx256"))]
 pub type GF2_128x8 = avx256::AVX256GF2_128x8;
 #[cfg(all(target_arch = "x86_64", not(feature = "avx256")))]
 pub type GF2_128x8 = avx::AVX512GF2_128x8;