Fix tests

Doridian · Aug 31, 2022 · 642ab2d · 642ab2d
1 parent df54b6a
commit 642ab2d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ipv4_test.go b/ipv4_test.go
@@ -46,14 +46,14 @@ readFrame:
 				if ethertype != waterutil.IPv4 {
 					continue readFrame
 				}
-				if expectBroadcast && !waterutil.IsBroadcast(waterutil.MACDestination(buffer)) {
+				if expectBroadcast && !waterutil.IsMACBroadcast(waterutil.MACDestination(buffer)) {
 					continue readFrame
 				}
 				packet = waterutil.MACPayload(buffer)
 			} else {
 				packet = buffer
 			}
-			if !waterutil.IsIPv4(packet) {
+			if waterutil.IPVersion(packet) != 4 {
 				continue readFrame
 			}
 			if !waterutil.IPv4Source(packet).Equal(expectSrc) {