diff --git a/test_utils/retry.py b/connection/utils/retry.py
similarity index 100%
rename from test_utils/retry.py
rename to connection/utils/retry.py
diff --git a/test_utils/os_utils.py b/test_utils/os_utils.py
index 4a39786..61dc2c3 100644
--- a/test_utils/os_utils.py
+++ b/test_utils/os_utils.py
@@ -21,7 +21,7 @@ from test_tools.disk_utils import get_sysfs_path
 from test_tools.fs_utils import check_if_directory_exists, create_directory, check_if_file_exists
 from test_utils.filesystem.file import File
 from connection.utils.output import CmdException
-from test_utils.retry import Retry
+from connection.utils.retry import Retry
 from types.size import Size, Unit
 
 DEBUGFS_MOUNT_POINT = "/sys/kernel/debug"