diff --git a/src/ptbench/engine/saliency/completeness.py b/src/ptbench/engine/saliency/completeness.py
index 87778d74955fe40d3f54882afd677949b08a6d7b..a75d57b2953c6cb283f96116e4e7d9e563fdce08 100644
--- a/src/ptbench/engine/saliency/completeness.py
+++ b/src/ptbench/engine/saliency/completeness.py
@@ -182,9 +182,9 @@ def run(
         * combined: ROAD combined score by evaluating ``(lerf-morf)/2``.
     """
 
-    from ..models.densenet import Densenet
-    from ..models.pasa import Pasa
-    from .saliencymap_generator import _create_saliency_map_callable
+    from ...models.densenet import Densenet
+    from ...models.pasa import Pasa
+    from .generator import _create_saliency_map_callable
 
     if isinstance(model, Pasa):
         if saliency_map_algorithm == "fullgrad":
diff --git a/src/ptbench/engine/saliency/generator.py b/src/ptbench/engine/saliency/generator.py
index eda78ed49937db7d718e048f93c20bd9eeabfd04..5d87167aa6234b7f72387923c95d826d100a126b 100644
--- a/src/ptbench/engine/saliency/generator.py
+++ b/src/ptbench/engine/saliency/generator.py
@@ -136,8 +136,8 @@ def run(
     """
     from pytorch_grad_cam.utils.model_targets import ClassifierOutputTarget
 
-    from ..models.densenet import Densenet
-    from ..models.pasa import Pasa
+    from ...models.densenet import Densenet
+    from ...models.pasa import Pasa
 
     if isinstance(model, Pasa):
         if "fullgrad" in saliency_map_algorithms: