diff --git a/test/dtypes/test_affine_quantized_float.py b/test/dtypes/test_affine_quantized_float.py
index 427a26969..34f6bd2f1 100644
--- a/test/dtypes/test_affine_quantized_float.py
+++ b/test/dtypes/test_affine_quantized_float.py
@@ -30,10 +30,10 @@
     float8_static_activation_float8_weight,
 )
 from torchao.quantization.quant_primitives import (
-    choose_qparams_affine,
     MappingType,
     PerRow,
     PerTensor,
+    choose_qparams_affine,
 )
 
 random.seed(0)