diff --git a/libraries/base/diag/tb/vhdl/tb_tb_mms_diag_block_gen.vhd b/libraries/base/diag/tb/vhdl/tb_tb_mms_diag_block_gen.vhd
index ffca8cfc6807eabd00404d2b9f23f73d9572ae93..9aa6982cbe213b0036ce3f46852515ccee8fe4ae 100644
--- a/libraries/base/diag/tb/vhdl/tb_tb_mms_diag_block_gen.vhd
+++ b/libraries/base/diag/tb/vhdl/tb_tb_mms_diag_block_gen.vhd
@@ -48,12 +48,12 @@ BEGIN
 -- g_nof_streams       : NATURAL := 16;
 -- g_gap_size          : NATURAL := 160
 
-  u_bg_one_stream  : ENTITY work.tb_mms_diag_block_gen GENERIC MAP (FALSE, TRUE, FALSE, FALSE, c_nof_repeat, 1,   0);
-  u_bg_gap_0       : ENTITY work.tb_mms_diag_block_gen GENERIC MAP (FALSE, TRUE, FALSE, FALSE, c_nof_repeat, 2,   0);
-  u_bg_gap_160     : ENTITY work.tb_mms_diag_block_gen GENERIC MAP (FALSE, TRUE, FALSE, FALSE, c_nof_repeat, 2, 160);
+  u_bg_one_stream  : ENTITY work.tb_mms_diag_block_gen GENERIC MAP (FALSE, TRUE, FALSE, FALSE, 1, 1,   0);
+  u_bg_gap_0       : ENTITY work.tb_mms_diag_block_gen GENERIC MAP (FALSE, TRUE, FALSE, FALSE, 1, 3,   0);
+  u_bg_gap_160     : ENTITY work.tb_mms_diag_block_gen GENERIC MAP (FALSE, TRUE, FALSE, FALSE, 1, 3, 160);
                                                                                              
-  u_mux_one_stream : ENTITY work.tb_mms_diag_block_gen GENERIC MAP ( TRUE, TRUE, FALSE, FALSE, c_nof_repeat, 1,   0);
-  u_mux_gap_0      : ENTITY work.tb_mms_diag_block_gen GENERIC MAP ( TRUE, TRUE, FALSE, FALSE, c_nof_repeat, 2,   1);
-  u_mux_gap_1      : ENTITY work.tb_mms_diag_block_gen GENERIC MAP ( TRUE, TRUE, FALSE,  TRUE, c_nof_repeat, 2,   1);
+  u_mux_one_stream : ENTITY work.tb_mms_diag_block_gen GENERIC MAP ( TRUE, TRUE, FALSE, FALSE, 5, 1,   0);
+  u_mux_gap_0      : ENTITY work.tb_mms_diag_block_gen GENERIC MAP ( TRUE, TRUE, FALSE, FALSE, 1, 2,   1);
+  u_mux_gap_1      : ENTITY work.tb_mms_diag_block_gen GENERIC MAP ( TRUE, TRUE, FALSE,  TRUE, 1, 2,   1);
   
 END tb;