diff --git a/src/lbaf/Execution/lbsBruteForceAlgorithm.py b/src/lbaf/Execution/lbsBruteForceAlgorithm.py
index 00e5e24d9..2ee926268 100644
--- a/src/lbaf/Execution/lbsBruteForceAlgorithm.py
+++ b/src/lbaf/Execution/lbsBruteForceAlgorithm.py
@@ -4,7 +4,6 @@
 from logging import Logger
 
 from .lbsAlgorithmBase import AlgorithmBase
-from ..Model.lbsPhase import Phase
 from ..Utils.exception_handler import exc_handler
 
 
@@ -75,7 +74,7 @@ def execute(self, p_id: int, phases: list, distributions: dict, statistics: dict
         objects = []
 
         # Iterate over ranks
-        phase_ranks = phases.get_ranks()
+        phase_ranks = phases[0].get_ranks()
         for rank in phase_ranks:
             for o in rank.get_objects():
                 entry = {