diff --git a/openvino_xai/inserter/__init__.py b/openvino_xai/inserter/__init__.py
index 3524a6e5..109e8a05 100644
--- a/openvino_xai/inserter/__init__.py
+++ b/openvino_xai/inserter/__init__.py
@@ -3,12 +3,12 @@
 """
 Interface for inserting XAI branch into OV IR.
 """
+from openvino_xai.inserter.inserter import insert_xai
 from openvino_xai.inserter.parameters import (
     ClassificationInsertionParameters,
     DetectionInsertionParameters,
     InsertionParameters,
 )
-from openvino_xai.inserter.inserter import insert_xai
 
 __all__ = [
     "insert_xai",
diff --git a/openvino_xai/inserter/inserter.py b/openvino_xai/inserter/inserter.py
index 365e6908..efbc240e 100644
--- a/openvino_xai/inserter/inserter.py
+++ b/openvino_xai/inserter/inserter.py
@@ -6,11 +6,11 @@
 
 from openvino_xai import Task
 from openvino_xai.common.utils import SALIENCY_MAP_OUTPUT_NAME, has_xai, logger
+from openvino_xai.inserter.parameters import InsertionParameters
 from openvino_xai.methods.white_box.create_method import (
     create_white_box_classification_explain_method,
     create_white_box_detection_explain_method,
 )
-from openvino_xai.inserter.parameters import InsertionParameters
 
 
 def insert_xai(