diff --git a/test/tb_floo_dma_mesh.sv b/test/tb_floo_dma_mesh.sv
index d7cd6710..5f936dc3 100644
--- a/test/tb_floo_dma_mesh.sv
+++ b/test/tb_floo_dma_mesh.sv
@@ -213,15 +213,14 @@ module tb_floo_dma_mesh;
       .WideReorderBufferSize    ( WideReorderBufferSize   ),
       .CutAx                    ( CutAx                   ),
       .CutRsp                   ( CutRsp                  ),
-      .xy_id_t                  ( xy_id_t                 )
+      .id_t                     ( xy_id_t                 )
     ) i_hbm_chimney [NumChimneys-1:0] (
       .clk_i                ( clk               ),
       .rst_ni               ( rst_n             ),
       .sram_cfg_i           ( '0                ),
       .test_enable_i        ( 1'b0              ),
-      .id_i                 ( '0                ),
+      .id_i                 ( xy_id_hbm         ),
       .id_map_i             ( '0                ),
-      .xy_id_i              ( xy_id_hbm         ),
       .axi_narrow_in_req_i  ( '0                ),
       .axi_narrow_in_rsp_o  (                   ),
       .axi_narrow_out_req_o ( narrow_hbm_req[i] ),
@@ -352,15 +351,14 @@ module tb_floo_dma_mesh;
         .WideReorderBufferSize    ( WideReorderBufferSize   ),
         .CutAx                    ( CutAx                   ),
         .CutRsp                   ( CutRsp                  ),
-        .xy_id_t                  ( xy_id_t                 )
+        .id_t                     ( xy_id_t                 )
       ) i_dma_chimney (
         .clk_i                ( clk                           ),
         .rst_ni               ( rst_n                         ),
         .sram_cfg_i           ( '0                            ),
         .test_enable_i        ( 1'b0                          ),
-        .id_i                 ( '0                            ),
+        .id_i                 ( current_id                    ),
         .id_map_i             ( '0                            ),
-        .xy_id_i              ( current_id                    ),
         .axi_narrow_in_req_i  ( narrow_man_req[x][y]          ),
         .axi_narrow_in_rsp_o  ( narrow_man_rsp[x][y]          ),
         .axi_narrow_out_req_o ( narrow_sub_req[x][y]          ),