diff --git a/pbrd/pbr_vty.c b/pbrd/pbr_vty.c
index bb09a10b98..c4e36ebd46 100644
--- a/pbrd/pbr_vty.c
+++ b/pbrd/pbr_vty.c
@@ -125,7 +125,7 @@ DEFPY(pbr_map_match_src, pbr_map_match_src_cmd,
 	if (!pbrms)
 		return CMD_WARNING_CONFIG_FAILED;
 
-	if (pbrms->dst && pbrms->family && prefix->family != pbrms->family) {
+	if (pbrms->dst && prefix->family != pbrms->dst->family) {
 		vty_out(vty, "Cannot mismatch families within match src/dst\n");
 		return CMD_WARNING_CONFIG_FAILED;
 	}
@@ -161,7 +161,7 @@ DEFPY(pbr_map_match_dst, pbr_map_match_dst_cmd,
 	if (!pbrms)
 		return CMD_WARNING_CONFIG_FAILED;
 
-	if (pbrms->src && pbrms->family && prefix->family != pbrms->family) {
+	if (pbrms->src && prefix->family != pbrms->src->family) {
 		vty_out(vty, "Cannot mismatch families within match src/dst\n");
 		return CMD_WARNING_CONFIG_FAILED;
 	}