diff --git a/dsv4/ops/quantize.py b/dsv4/ops/quantize.py
index ecdf6544..798b19de 100644
--- a/dsv4/ops/quantize.py
+++ b/dsv4/ops/quantize.py
@@ -5,6 +5,7 @@ import cutlass
 import cutlass.cute as cute
 import cutlass.torch as cutlass_torch
 import cutlass.utils as utils
+from dsv4.ops.layouts import ceil_div
 
 from dsv4.kernels.gemm.grouped import (
     cat_byte_reinterpretable_tensors,