diff --git a/test/functional/api/cas/cache.py b/test/functional/api/cas/cache.py
index 460010a..e752a0f 100644
--- a/test/functional/api/cas/cache.py
+++ b/test/functional/api/cas/cache.py
@@ -194,7 +194,7 @@ class Cache:
     def set_params_nhit(self, promotion_params_nhit: PromotionParametersNhit) -> Output:
         return casadm.set_param_promotion_nhit(
             self.cache_id,
-            threshold=promotion_params_nhit.threshold.get_value(),
+            threshold=promotion_params_nhit.threshold,
             trigger=promotion_params_nhit.trigger
         )
 
diff --git a/test/functional/api/cas/cache_config.py b/test/functional/api/cas/cache_config.py
index 7bb5d6f..128d4dc 100644
--- a/test/functional/api/cas/cache_config.py
+++ b/test/functional/api/cas/cache_config.py
@@ -240,7 +240,7 @@ class SeqCutOffParameters:
 
 
 class PromotionParametersNhit:
-    def __init__(self, threshold: Size = None, trigger: int = None):
+    def __init__(self, threshold: int = None, trigger: int = None):
         self.threshold = threshold
         self.trigger = trigger