diff --git a/tests/test_pipeline_real_weights.py b/tests/test_pipeline_real_weights.py
index 0fa05b11..244218e1 100644
--- a/tests/test_pipeline_real_weights.py
+++ b/tests/test_pipeline_real_weights.py
@@ -184,9 +184,10 @@ def main():
     # ---- Reference: same runner but with dynamic gs (quantize_to_nvfp4) ----
     print("\n--- Reference (dynamic gs via quantize_to_nvfp4) ---")
     # We'll use the same runner infrastructure but manually call the reference path
-    from cutedsl.quantize import quantize_to_nvfp4
-    from cutedsl.gemm import run_nvfp4_grouped_gemm
-    from cutedsl.bridge import assemble_scales_3d_side, make_b_k_major, assemble_scales_3d_side
+    from cutedsl.bridge import (
+        quantize_to_nvfp4, run_nvfp4_grouped_gemm,
+        assemble_scales_3d_side, make_b_k_major,
+    )
     
     with torch.no_grad():
         # Stack weights for GEMM