diff --git a/src/vnet/bier/bier_test.c b/src/vnet/bier/bier_test.c
index 2c13abc..08a8c55 100644
--- a/src/vnet/bier/bier_test.c
+++ b/src/vnet/bier/bier_test.c
@@ -158,7 +158,7 @@
 
 static int
 bier_test_validate_entry (index_t bei,
-                          u16 n_buckets,
+                          int n_buckets,
                           ...)
 {
     dpo_id_t dpo = DPO_INVALID;
diff --git a/src/vnet/fib/fib_test.c b/src/vnet/fib/fib_test.c
index 61b290b..2a30b3c 100644
--- a/src/vnet/fib/fib_test.c
+++ b/src/vnet/fib/fib_test.c
@@ -355,7 +355,7 @@
 
 int
 fib_test_validate_lb_v (const load_balance_t *lb,
-			u16 n_buckets,
+			int n_buckets,
 			va_list *ap)
 {
     const dpo_id_t *dpo;
@@ -603,7 +603,7 @@
 
 int
 fib_test_validate_lb (const dpo_id_t *dpo,
-                     u16 n_buckets,
+                     int n_buckets,
                      ...)
 {
     const load_balance_t *lb;
diff --git a/src/vnet/fib/fib_test.h b/src/vnet/fib/fib_test.h
index 53697cb..0309b3f 100644
--- a/src/vnet/fib/fib_test.h
+++ b/src/vnet/fib/fib_test.h
@@ -118,11 +118,11 @@
                                    va_list *ap);
 
 extern int fib_test_validate_lb_v(const load_balance_t *lb,
-                                  u16 n_buckets,
+                                  int n_buckets,
                                   va_list *ap);
 
 extern int fib_test_validate_lb(const dpo_id_t *dpo,
-				u16 n_buckets,
+				int n_buckets,
 				...);
 
 extern int fib_test_validate_entry(fib_node_index_t fei,