diff --git a/tests/unit/Execution/test_lbs_clustering_transfer_strategy.py b/tests/unit/Execution/test_lbs_clustering_transfer_strategy.py
index 2ebb7a2a..66fde953 100644
--- a/tests/unit/Execution/test_lbs_clustering_transfer_strategy.py
+++ b/tests/unit/Execution/test_lbs_clustering_transfer_strategy.py
@@ -84,7 +84,6 @@ def setUp(self):
 
         # Define the rank and declare known peers
         self.rank = Rank(r_id=0, migratable_objects=self.migratable_objects, sentinel_objects=self.sentinel_objects, logger=self.logger)
-        self.rank.set_shared_blocks(self.block_set)
         self.known_peers = {}
 
        # Instantiate the phase
@@ -176,10 +175,6 @@ def test_lbs_clustering_transfer_strategy_iterate_subclusters(self):
         rank0 = Rank(r_id=r_id0, migratable_objects={obj0, obj1, obj2}, logger=self.logger)
         rank1 = Rank(r_id=r_id1, migratable_objects={obj3, obj4, obj5}, logger=self.logger)
 
-        # Assign ranks to memory blocks
-        rank0.set_shared_blocks({block0, block1})
-        rank1.set_shared_blocks({block2, block3})
-
         # Create known_peers set
         rank_list = [rank0, rank1]
         known_peers = {}