diff --git a/test_gemm_1group.py b/test_gemm_1group.py
index e75698be..1e3b7147 100644
--- a/test_gemm_1group.py
+++ b/test_gemm_1group.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 """Test: run_nvfp4_grouped_gemm with num_groups=1 on different GPUs."""
 import torch
-from dsv4.ops.gemm_runner import run_nvfp4_grouped_gemm, warmup_nvfp4_compilation
+from dsv4.ops.gemm_runner import run_nvfp4_grouped_gemm
 from dsv4.ops.quantize import quantize_nvfp4_gpu
 from dsv4.ops.layouts import make_b_k_major, assemble_scales_3d_side