diff --git a/tests/intg/test_detection.py b/tests/intg/test_detection.py
index d9970d31..96e9eb34 100644
--- a/tests/intg/test_detection.py
+++ b/tests/intg/test_detection.py
@@ -14,7 +14,7 @@
 from openvino_xai.common.utils import retrieve_otx_model
 from openvino_xai.explainer.explainer import Explainer, ExplainMode
 from openvino_xai.explainer.utils import get_preprocess_fn
-from openvino_xai.methods.black_box.aise import AISEDetection
+from openvino_xai.methods.black_box.aise.detection import AISEDetection
 from openvino_xai.methods.factory import BlackBoxMethodFactory, WhiteBoxMethodFactory
 from openvino_xai.methods.white_box.det_class_probability_map import (
     DetClassProbabilityMap,
diff --git a/tests/unit/methods/black_box/test_black_box_method.py b/tests/unit/methods/black_box/test_black_box_method.py
index 7f889e18..c9b48e68 100644
--- a/tests/unit/methods/black_box/test_black_box_method.py
+++ b/tests/unit/methods/black_box/test_black_box_method.py
@@ -11,7 +11,8 @@
 
 from openvino_xai.common.utils import retrieve_otx_model
 from openvino_xai.explainer.utils import get_postprocess_fn, get_preprocess_fn
-from openvino_xai.methods.black_box.aise import AISEClassification, AISEDetection
+from openvino_xai.methods.black_box.aise.classification import AISEClassification
+from openvino_xai.methods.black_box.aise.detection import AISEDetection
 from openvino_xai.methods.black_box.base import Preset
 from openvino_xai.methods.black_box.rise import RISE
 from tests.intg.test_classification import DEFAULT_CLS_MODEL
diff --git a/tests/unit/methods/test_factory.py b/tests/unit/methods/test_factory.py
index 89abaf79..b197c1c2 100644
--- a/tests/unit/methods/test_factory.py
+++ b/tests/unit/methods/test_factory.py
@@ -10,7 +10,7 @@
 from openvino_xai.common.parameters import Method, Task
 from openvino_xai.common.utils import retrieve_otx_model
 from openvino_xai.explainer.utils import get_postprocess_fn, get_preprocess_fn
-from openvino_xai.methods.black_box.aise import AISEClassification
+from openvino_xai.methods.black_box.aise.classification import AISEClassification
 from openvino_xai.methods.factory import BlackBoxMethodFactory, WhiteBoxMethodFactory
 from openvino_xai.methods.white_box.activation_map import ActivationMap
 from openvino_xai.methods.white_box.det_class_probability_map import (