diff --git a/test/functional/api/cas/cache_config.py b/test/functional/api/cas/cache_config.py
index 4cb9100..c66e862 100644
--- a/test/functional/api/cas/cache_config.py
+++ b/test/functional/api/cas/cache_config.py
@@ -24,7 +24,7 @@ class CacheLineSize(Enum):
 class CacheModeTrait(IntFlag):
     InsertWrite = 1
     InsertRead = 2
-    LazyFlush = 4
+    LazyWrites = 4
 
 
 class CacheMode(Enum):
@@ -45,9 +45,11 @@ class CacheMode(Enum):
         elif cache_mode == CacheMode.WT:
             return CacheModeTrait.InsertRead | CacheModeTrait.InsertWrite
         elif cache_mode == CacheMode.WB:
-            return CacheModeTrait.InsertRead | CacheModeTrait.InsertWrite | CacheModeTrait.LazyFlush
+            return (
+                CacheModeTrait.InsertRead | CacheModeTrait.InsertWrite | CacheModeTrait.LazyWrites
+            )
         elif cache_mode == CacheMode.WO:
-            return CacheModeTrait.InsertWrite | CacheModeTrait.LazyFlush
+            return CacheModeTrait.InsertWrite | CacheModeTrait.LazyWrites
         elif cache_mode == CacheMode.WA:
             return CacheModeTrait.InsertRead
 
diff --git a/test/functional/tests/io/test_write_fetch.py b/test/functional/tests/io/test_write_fetch.py
index d3ea6ab..2d3cbb4 100644
--- a/test/functional/tests/io/test_write_fetch.py
+++ b/test/functional/tests/io/test_write_fetch.py
@@ -153,7 +153,7 @@ def check_io_stats(cache_disk, cache, io_stats_before, io_size, blocksize, skip_
     expected_writes = io_size * (blocksize / (blocksize + skip_size))
 
     cache_mode_traits = CacheMode.get_traits(cache.get_cache_mode())
-    if CacheModeTrait.InsertWrite | CacheModeTrait.LazyFlush in cache_mode_traits:
+    if CacheModeTrait.InsertWrite | CacheModeTrait.LazyWrites in cache_mode_traits:
         # Metadata size is 4KiB per each cache line
         metadata_size = (io_size / cache.get_cache_line_size().value) * Size(4, Unit.KibiByte)
         expected_writes += metadata_size