diff --git a/bungee/src/main/java/com/viaversion/viaversion/bungee/handlers/BungeeServerHandler.java b/bungee/src/main/java/com/viaversion/viaversion/bungee/handlers/BungeeServerHandler.java
index 67cf34f1c..25b039ae2 100644
--- a/bungee/src/main/java/com/viaversion/viaversion/bungee/handlers/BungeeServerHandler.java
+++ b/bungee/src/main/java/com/viaversion/viaversion/bungee/handlers/BungeeServerHandler.java
@@ -283,9 +283,9 @@ public class BungeeServerHandler implements Listener {
         }
 
         Object wrapper = channelWrapper.get(player);
-        setVersion.invoke(wrapper, serverProtocolVersion);
+        setVersion.invoke(wrapper, serverProtocolVersion.getVersion());
 
-        Object entityMap = getEntityMap.invoke(null, serverProtocolVersion);
+        Object entityMap = getEntityMap.invoke(null, serverProtocolVersion.getVersion());
         entityRewrite.set(player, entityMap);
     }
 }