From 766048b1eb99b00abdaee62f773998cbefdd8eb4 Mon Sep 17 00:00:00 2001
From: RaphiMC <50594595+RaphiMC@users.noreply.github.com>
Date: Mon, 27 Nov 2023 17:56:08 +0100
Subject: [PATCH] Fixed connection session setup

---
 .../java/de/florianmichael/viafabricplus/ViaFabricPlus.java    | 1 -
 .../mixin/fixes/minecraft/screen/MixinConnectScreen_1.java     | 3 +--
 2 files changed, 1 insertion(+), 3 deletions(-)

diff --git a/src/main/java/de/florianmichael/viafabricplus/ViaFabricPlus.java b/src/main/java/de/florianmichael/viafabricplus/ViaFabricPlus.java
index b1a43ad4..1ad5a3b8 100644
--- a/src/main/java/de/florianmichael/viafabricplus/ViaFabricPlus.java
+++ b/src/main/java/de/florianmichael/viafabricplus/ViaFabricPlus.java
@@ -51,7 +51,6 @@ import java.util.concurrent.CompletableFuture;
  *  - Add setting for VFP AlphaInventoryProvider
  *  - Fix MixinAbstractDonkeyEntity
  *  - Check TO DO in MixinEntity
- *  - Fix bedrock online mode
  */
 public class ViaFabricPlus {
     private static final ViaFabricPlus instance = new ViaFabricPlus();
diff --git a/src/main/java/de/florianmichael/viafabricplus/injection/mixin/fixes/minecraft/screen/MixinConnectScreen_1.java b/src/main/java/de/florianmichael/viafabricplus/injection/mixin/fixes/minecraft/screen/MixinConnectScreen_1.java
index 806bac7c..61ac57ef 100644
--- a/src/main/java/de/florianmichael/viafabricplus/injection/mixin/fixes/minecraft/screen/MixinConnectScreen_1.java
+++ b/src/main/java/de/florianmichael/viafabricplus/injection/mixin/fixes/minecraft/screen/MixinConnectScreen_1.java
@@ -75,8 +75,7 @@ public abstract class MixinConnectScreen_1 {
         final UserConnection userConnection = ((IClientConnection) clientConnection).viaFabricPlus$getUserConnection();
         if (userConnection == null) return;
 
-        final VersionEnum targetVersion = VersionEnum.fromUserConnection(userConnection);
-
+        final VersionEnum targetVersion = ProtocolHack.getTargetVersion();
         if (targetVersion.isBetweenInclusive(VersionEnum.r1_19, VersionEnum.r1_19_1tor1_19_2)) {
             final var keyPair = MinecraftClient.getInstance().getProfileKeys().fetchKeyPair().join().orElse(null);
             if (keyPair != null) {