diff --git a/hw/tb/tb_floo_axi_mesh.sv b/hw/tb/tb_floo_axi_mesh.sv
index 4ccd1631..c479e051 100644
--- a/hw/tb/tb_floo_axi_mesh.sv
+++ b/hw/tb/tb_floo_axi_mesh.sv
@@ -93,7 +93,7 @@ module tb_floo_axi_mesh;
       localparam string DmaName = $sformatf("dma_%0d_%0d", x, y);
 
       localparam int unsigned Index = x * NumY + y;
-      localparam addr_t MemBaseAddr = Sam[ClusterNi00+Index].start_addr;
+      localparam addr_t MemBaseAddr = Sam[ClusterX0Y0+Index].start_addr;
 
       floo_dma_test_node #(
         .TA             ( ApplTime                                  ),
diff --git a/hw/tb/tb_floo_nw_mesh.sv b/hw/tb/tb_floo_nw_mesh.sv
index ffbec10b..0daf4dcf 100644
--- a/hw/tb/tb_floo_nw_mesh.sv
+++ b/hw/tb/tb_floo_nw_mesh.sv
@@ -133,7 +133,7 @@ module tb_floo_nw_mesh;
       localparam string WideDmaName   = $sformatf("wide_dma_%0d_%0d", x, y);
 
       localparam int unsigned Index = x * NumY + y;
-      localparam addr_t MemBaseAddr = Sam[ClusterNi00+Index].start_addr;
+      localparam addr_t MemBaseAddr = Sam[ClusterX0Y0+Index].start_addr;
 
       floo_dma_test_node #(
         .TA             ( ApplTime                                  ),