diff --git a/Assignment 4/src/ca/cosc3p91/a4/util/network/Client.java b/Assignment 4/src/ca/cosc3p91/a4/util/network/Client.java
index 8b1f9d8..35d26c6 100644
--- a/Assignment 4/src/ca/cosc3p91/a4/util/network/Client.java	
+++ b/Assignment 4/src/ca/cosc3p91/a4/util/network/Client.java	
@@ -110,7 +110,7 @@ public class Client implements Runnable {
     }
 
     private void sendMessage(Message.Sent message){
-        if (message.getMessageID() != PacketTable.ACK)
+        if (message.getPacketID() != PacketTable.ACK)
             sentMessages.put(message.getMessageID(), message);
         byte[] data = message.getData().toByteArray();
         DatagramPacket sendPacket = new DatagramPacket(data, data.length, serverAddress, Server.SERVER_PORT);
diff --git a/Assignment 4/src/ca/cosc3p91/a4/util/network/Server.java b/Assignment 4/src/ca/cosc3p91/a4/util/network/Server.java
index 2436141..6121661 100644
--- a/Assignment 4/src/ca/cosc3p91/a4/util/network/Server.java	
+++ b/Assignment 4/src/ca/cosc3p91/a4/util/network/Server.java	
@@ -176,7 +176,7 @@ public class Server implements Runnable {
         }
 
         public void sendMessage(Message.Sent message){
-            if (message.getMessageID() != PacketTable.ACK)
+            if (message.getPacketID() != PacketTable.ACK)
                 this.sentMessages.put(message.getMessageID(), message);
             byte[] data = message.getData().toByteArray();
             if (data.length > PACKET_SIZE)