diff --git a/main.go b/main.go
index cc95037..15dd8e6 100644
--- a/main.go
+++ b/main.go
@@ -50,7 +50,7 @@ func testmain() {
 	}()
 
 	config := water.Config{
-		DeviceType: water.TAP,
+		DeviceType: water.TUN,
 	}
 	config.Name = "stun"
 
@@ -70,7 +70,7 @@ func testmain() {
 		frame = frame[:n]
 
 		log.Printf("DstMac: %s SrcMac: %s\n", waterutil.MACDestination(frame), waterutil.MACSource(frame))
-		log.Printf("Dst: %s Src: %s\n", frame.Destination(), frame.Source())
+		log.Printf("DstIP: %s SrcIP: %s\n", waterutil.IPv4Destination(frame), waterutil.IPv4Source(frame))
 		log.Printf("Ethertype: % x\n", frame.Ethertype())
 		// log.Printf("Payload: % x\n", frame.Payload())