diff --git a/tb/test_axis_arb_mux_4.py b/tb/test_axis_arb_mux_4.py
index 77e52d35e..cd5489475 100755
--- a/tb/test_axis_arb_mux_4.py
+++ b/tb/test_axis_arb_mux_4.py
@@ -55,6 +55,8 @@ def bench():
     DEST_WIDTH = 8
     USER_ENABLE = 1
     USER_WIDTH = 1
+    ARB_TYPE = "PRIORITY"
+    LSB_PRIORITY = "HIGH"
 
     # Inputs
     clk = Signal(bool(0))
diff --git a/tb/test_axis_arb_mux_4.v b/tb/test_axis_arb_mux_4.v
index 79c4f2736..ae929053d 100644
--- a/tb/test_axis_arb_mux_4.v
+++ b/tb/test_axis_arb_mux_4.v
@@ -42,6 +42,8 @@ parameter DEST_ENABLE = 1;
 parameter DEST_WIDTH = 8;
 parameter USER_ENABLE = 1;
 parameter USER_WIDTH = 1;
+parameter ARB_TYPE = "PRIORITY";
+parameter LSB_PRIORITY = "HIGH";
 
 // Inputs
 reg clk = 0;
@@ -110,7 +112,9 @@ axis_arb_mux #(
     .DEST_ENABLE(DEST_ENABLE),
     .DEST_WIDTH(DEST_WIDTH),
     .USER_ENABLE(USER_ENABLE),
-    .USER_WIDTH(USER_WIDTH)
+    .USER_WIDTH(USER_WIDTH),
+    .ARB_TYPE(ARB_TYPE),
+    .LSB_PRIORITY(LSB_PRIORITY)
 )
 UUT (
     .clk(clk),
diff --git a/tb/test_axis_arb_mux_4_64.py b/tb/test_axis_arb_mux_4_64.py
index f17ba4ec2..b462ce9f1 100755
--- a/tb/test_axis_arb_mux_4_64.py
+++ b/tb/test_axis_arb_mux_4_64.py
@@ -55,6 +55,8 @@ def bench():
     DEST_WIDTH = 8
     USER_ENABLE = 1
     USER_WIDTH = 1
+    ARB_TYPE = "PRIORITY"
+    LSB_PRIORITY = "HIGH"
 
     # Inputs
     clk = Signal(bool(0))
diff --git a/tb/test_axis_arb_mux_4_64.v b/tb/test_axis_arb_mux_4_64.v
index eaa9c96d8..107a4de12 100644
--- a/tb/test_axis_arb_mux_4_64.v
+++ b/tb/test_axis_arb_mux_4_64.v
@@ -42,6 +42,8 @@ parameter DEST_ENABLE = 1;
 parameter DEST_WIDTH = 8;
 parameter USER_ENABLE = 1;
 parameter USER_WIDTH = 1;
+parameter ARB_TYPE = "PRIORITY";
+parameter LSB_PRIORITY = "HIGH";
 
 // Inputs
 reg clk = 0;
@@ -110,7 +112,9 @@ axis_arb_mux #(
     .DEST_ENABLE(DEST_ENABLE),
     .DEST_WIDTH(DEST_WIDTH),
     .USER_ENABLE(USER_ENABLE),
-    .USER_WIDTH(USER_WIDTH)
+    .USER_WIDTH(USER_WIDTH),
+    .ARB_TYPE(ARB_TYPE),
+    .LSB_PRIORITY(LSB_PRIORITY)
 )
 UUT (
     .clk(clk),