diff --git a/src/main/java/edu/byu/cs/controller/WebSocketController.java b/src/main/java/edu/byu/cs/controller/WebSocketController.java
index 88f72b8c..76731c02 100644
--- a/src/main/java/edu/byu/cs/controller/WebSocketController.java
+++ b/src/main/java/edu/byu/cs/controller/WebSocketController.java
@@ -32,7 +32,7 @@ public static void onMessage(WsMessageContext ctx) {
         String netId;
         ctx.enableAutomaticPings(20, TimeUnit.SECONDS);
         try {
-            netId = JwtUtils.validateToken(message);
+            netId = JwtUtils.validateToken(ctx.cookie("token"));
         } catch (Exception e) {
             LOGGER.warn("Exception thrown while validating token: ", e);
             sendError(session, "Invalid token");