diff --git a/src/main/java/earth/krakatao/protocol/KraSocketClientProtocolMessage.java b/src/main/java/earth/krakatao/protocol/KraSocketClientProtocolMessage.java
index 76f3980..a93cbb9 100644
--- a/src/main/java/earth/krakatao/protocol/KraSocketClientProtocolMessage.java
+++ b/src/main/java/earth/krakatao/protocol/KraSocketClientProtocolMessage.java
@@ -20,12 +20,12 @@ public class KraSocketClientProtocolMessage {
   private long time;
 
   public KraSocketClientProtocolMessage(byte status, int cmdID, int dest, String uuid,
-      short cmdNumber, String args, Consumer<KraSocketClientProtocolMessage> messageConsumer) {
+      int cmdNumber, String args, Consumer<KraSocketClientProtocolMessage> messageConsumer) {
     this.status = status;
     this.cmdID = cmdID;
     this.dest = dest;
     this.uuid = uuid;
-    this.cmdNumber = cmdNumber;
+    this.cmdNumber = (short) cmdNumber;
     this.args = args;
     this.messageConsumer = messageConsumer;