diff --git a/nncf/onnx/graph/model_utils.py b/nncf/onnx/graph/model_utils.py
index 62facd50614..41c9afcfe54 100644
--- a/nncf/onnx/graph/model_utils.py
+++ b/nncf/onnx/graph/model_utils.py
@@ -28,6 +28,7 @@ def remove_fq_from_inputs(model: onnx.ModelProto, nncf_graph: NNCFGraph) -> onnx
     It's needed for the further bias shift calculation that relates on quantized weights.
 
     :param model: onnx.ModelProto instance.
+    :param nncf_graph: NNCFGraph instance.
     :return: onnx.ModelProto instance without activation Quantizer nodes.
     """
     transformation_layout = TransformationLayout()
diff --git a/nncf/openvino/graph/model_utils.py b/nncf/openvino/graph/model_utils.py
index aac584c321b..803b72aa811 100644
--- a/nncf/openvino/graph/model_utils.py
+++ b/nncf/openvino/graph/model_utils.py
@@ -56,6 +56,7 @@ def remove_fq_from_inputs(model: ov.Model, graph: NNCFGraph) -> ov.Model:
     It's needed for the further bias shift calculation that relates on quantized weights.
 
     :param model: ov.Model instance.
+    :param graph: NNCFGraph instance.
     :return: ov.Model instance without activation Fake Quantize nodes.
     """
     transformation_layout = TransformationLayout()