diff --git a/tests/test_mirror.py b/tests/test_mirror.py
index c6e88ec557..74ef22e693 100644
--- a/tests/test_mirror.py
+++ b/tests/test_mirror.py
@@ -60,7 +60,7 @@ def remove_ip_address(self, interface, ip):
     def add_neighbor(self, interface, ip, mac):
         tbl = swsscommon.ProducerStateTable(self.pdb, "NEIGH_TABLE")
         fvs = swsscommon.FieldValuePairs([("neigh", mac),
-                                          ("family", "IPv4")])
+                                          ("family", "IPv4" if "." in ip else "IPv6")])
         tbl.set(interface + ":" + ip, fvs)
         time.sleep(1)