diff --git a/test/functional/test-framework b/test/functional/test-framework
index 7512420..6970a43 160000
--- a/test/functional/test-framework
+++ b/test/functional/test-framework
@@ -1 +1 @@
-Subproject commit 7512420e2adbecf2a83847f963f61326d17c77fc
+Subproject commit 6970a4310790fc56023661a5c7c1bddca714ceba
diff --git a/test/functional/tests/cli/test_zero_metadata_command.py b/test/functional/tests/cli/test_zero_metadata_command.py
index e33eee2..efb1ee1 100644
--- a/test/functional/tests/cli/test_zero_metadata_command.py
+++ b/test/functional/tests/cli/test_zero_metadata_command.py
@@ -15,7 +15,7 @@ from storage_devices.disk import DiskTypeSet, DiskType, DiskTypeLowerThan
 from test_tools.disk_utils import get_device_filesystem_type, Filesystem
 from test_tools.fio.fio import Fio
 from test_tools.fio.fio_param import IoEngine, ReadWrite
-from test_utils.disk_finder import get_system_disks
+from test_tools.disk_finder import get_system_disks
 from connection.utils.output import CmdException
 from types.size import Size, Unit
 
diff --git a/test/functional/tests/conftest.py b/test/functional/tests/conftest.py
index b5f7efd..648362e 100644
--- a/test/functional/tests/conftest.py
+++ b/test/functional/tests/conftest.py
@@ -25,7 +25,6 @@ from storage_devices.raid import Raid
 from storage_devices.ramdisk import RamDisk
 from test_tools.os_tools import kill_all_io
 from test_tools.udev import Udev
-from test_utils.disk_finder import get_disk_serial_number
 from test_tools.disk_utils import PartitionTable, create_partition_table
 from test_tools.device_mapper import DeviceMapper
 from test_tools.mdadm import Mdadm
@@ -34,6 +33,7 @@ from test_tools import initramfs, git
 from log.logger import create_log, Log
 from test_utils.common.singleton import Singleton
 from storage_devices.lvm import Lvm, LvmConfiguration
+from storage_devices.disk import Disk
 
 
 def pytest_addoption(parser):
@@ -185,7 +185,7 @@ def base_prepare(item):
 
         RamDisk.remove_all()
         for disk in TestRun.disks.values():
-            disk_serial = get_disk_serial_number(disk.path)
+            disk_serial = Disk.get_disk_serial_number(disk.path)
             if disk.serial_number and disk.serial_number != disk_serial:
                 raise Exception(
                     f"Serial for {disk.path} doesn't match the one from the config."