X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=test%2Fvpp_neighbor.py;fp=test%2Fvpp_neighbor.py;h=ffe87d93b586655ae8f6cf851767f43a01feba65;hb=e64e5fff4ddea88f386657c5d95ae8dc78138d20;hp=6172d8f871df0d93656cdc848981194492dc568f;hpb=58db6e16cf4f3bb1740ca2f62d7d887baad58d63;p=vpp.git

diff --git a/test/vpp_neighbor.py b/test/vpp_neighbor.py
index 6172d8f871d..ffe87d93b58 100644
--- a/test/vpp_neighbor.py
+++ b/test/vpp_neighbor.py
@@ -16,11 +16,8 @@ except NameError:
 def find_nbr(test, sw_if_index, nbr_addr, is_static=0, mac=None):
     ip_addr = ip_address(text_type(nbr_addr))
     e = VppEnum.vl_api_ip_neighbor_flags_t
-    if 6 == ip_addr.version:
-        af = VppEnum.vl_api_address_family_t.ADDRESS_IP6
-    else:
-        af = VppEnum.vl_api_address_family_t.ADDRESS_IP4
-    nbrs = test.vapi.ip_neighbor_dump(sw_if_index=sw_if_index, af=af)
+    nbrs = test.vapi.ip_neighbor_dump(sw_if_index=sw_if_index,
+                                      af=ip_addr.vapi_af)
 
     for n in nbrs:
         if ip_addr == n.neighbor.ip_address and \