diff --git a/models/export.py b/models/export.py
index 278fb476e783..6043c84c6246 100644
--- a/models/export.py
+++ b/models/export.py
@@ -18,7 +18,7 @@
 from torch.utils.mobile_optimizer import optimize_for_mobile
 
 from sparseml.pytorch.utils import ModuleExporter
-from sparseml.pytorch.utils.quantization import skip_onnx_input_quantize
+from sparseml.pytorch.sparsification.quantization import skip_onnx_input_quantize
 
 import models
 from models.experimental import attempt_load