diff --git a/tests/sparseml/transformers/obcq/test_mask_structure_preservation.py b/tests/sparseml/transformers/obcq/test_mask_structure_preservation.py
index eca6f5d237..358213cf21 100644
--- a/tests/sparseml/transformers/obcq/test_mask_structure_preservation.py
+++ b/tests/sparseml/transformers/obcq/test_mask_structure_preservation.py
@@ -19,7 +19,7 @@
 import pytest
 
 import sparseml
-from compressed_tensors.compressors.utils import tensor_follows_mask_structure
+from compressed_tensors.utils import tensor_follows_mask_structure
 from parameterized import parameterized_class
 from tests.testing_utils import parse_params, requires_torch