X-Git-Url: https://gerrit.fd.io/r/gitweb?p=vpp.git;a=blobdiff_plain;f=src%2Fplugins%2Funittest%2Fpunt_test.c;h=0c4622283e02f579c06b8a85588ec2b7338655b3;hp=431624fbea0ec357143b37a2212fcc822c4fbbbf;hb=45723b8d3;hpb=4e3f7b2869925b0812a58d04c4bf6371e6773630

diff --git a/src/plugins/unittest/punt_test.c b/src/plugins/unittest/punt_test.c
index 431624fbea0..0c4622283e0 100644
--- a/src/plugins/unittest/punt_test.c
+++ b/src/plugins/unittest/punt_test.c
@@ -18,6 +18,7 @@
 #include <vnet/ethernet/ethernet.h>
 #include <vnet/adj/adj.h>
 #include <vnet/ip/ip.h>
+#include <vnet/ip/punt.h>
 
 typedef enum punt_next_t_
 {
@@ -316,12 +317,12 @@ punt_test (vlib_main_t * vm,
 	   */
 	  punt_hdl = vlib_punt_client_register ("test");
 
-	  rc =
-	    vlib_punt_reason_alloc (punt_hdl, "reason-v4",
-				    NULL, NULL, &punt_reason_v4);
-	  rc |=
-	    vlib_punt_reason_alloc (punt_hdl, "reason-v6",
-				    NULL, NULL, &punt_reason_v6);
+	  rc = vlib_punt_reason_alloc (
+	    punt_hdl, "reason-v4", NULL, NULL, &punt_reason_v4,
+	    VNET_PUNT_REASON_F_IP4_PACKET, format_vnet_punt_reason_flags);
+	  rc |= vlib_punt_reason_alloc (
+	    punt_hdl, "reason-v6", NULL, NULL, &punt_reason_v6,
+	    VNET_PUNT_REASON_F_IP6_PACKET, format_vnet_punt_reason_flags);
 	  ASSERT (!rc);
 
 	  vnet_feature_enable_disable ("ip4-unicast",