diff --git a/test/functional/api/cas/cache_config.py b/test/functional/api/cas/cache_config.py
index de97969..5936914 100644
--- a/test/functional/api/cas/cache_config.py
+++ b/test/functional/api/cas/cache_config.py
@@ -4,10 +4,10 @@
 #
 
 from aenum import Enum, IntFlag
-from attotime import attotimedelta
 
 from test_utils.os_utils import get_kernel_module_parameter
 from test_utils.size import Size, Unit
+from test_utils.time import Time
 
 
 class CacheLineSize(Enum):
@@ -130,11 +130,6 @@ class CacheStatus(Enum):
         return self.value
 
 
-class Time(attotimedelta):
-    def total_milliseconds(self):
-        return int(self.total_seconds() * 1000)
-
-
 class FlushParametersAlru:
     def __init__(
         self,