diff --git a/p4src/nat64.p4 b/p4src/nat64.p4
index 79163e3..02b88b0 100644
--- a/p4src/nat64.p4
+++ b/p4src/nat64.p4
@@ -236,9 +236,9 @@ Echo or Echo Reply Message
 
     /* automatic translations */
     action nat64_tcp_session_translate(
-        ipv4_t src_addr,
+        ipv4_addr_t src_addr,
         bit<16> src_port,
-        ipv4_t dst_addr,
+        ipv4_addr_t dst_addr,
         bit<16> dst_port)
     {
 		hdr.ipv4.setValid();
@@ -250,9 +250,9 @@ Echo or Echo Reply Message
     }
 
     action nat46_tcp_session_translate(
-        ipv6_t src_addr,
+        ipv6_addr_t src_addr,
         bit<16> src_port,
-        ipv6_t dst_addr,
+        ipv6_addr_t dst_addr,
         bit<16> dst_port)
     {
 		hdr.ipv6.setValid();