diff --git a/tests/deepsparse/utils/onnx.py b/tests/deepsparse/utils/onnx.py
index ffe6499845..34b95c9a47 100644
--- a/tests/deepsparse/utils/onnx.py
+++ b/tests/deepsparse/utils/onnx.py
@@ -16,7 +16,7 @@
 import onnx
 
 import pytest
-from deepsparse.utils import override_onnx_batch_size, override_onnx_input_shapes
+from deepsparse.utils.onnx import override_onnx_batch_size, override_onnx_input_shapes
 from sparsezoo import Model