diff --git a/tests/sweep_framework/sweeps/eltwise/unary_backward/hardshrink_bw.py b/tests/sweep_framework/sweeps/eltwise/unary_backward/hardshrink_bw.py
index 50daf64e03d..12f14f69fa9 100644
--- a/tests/sweep_framework/sweeps/eltwise/unary_backward/hardshrink_bw.py
+++ b/tests/sweep_framework/sweeps/eltwise/unary_backward/hardshrink_bw.py
@@ -8,7 +8,7 @@
 import torch
 import random
 import ttnn
-from tests.sweep_framework.utils import gen_shapes
+from tests.sweep_framework.sweep_utils.utils import gen_shapes
 from tests.tt_eager.python_api_testing.sweep_tests.generation_funcs import gen_func_with_cast_tt
 
 from tests.ttnn.utils_for_testing import check_with_pcc, start_measuring_time, stop_measuring_time