Cleanup

Author: Exterminate5573

src/main/java/com/viaversion/viafabricplus/injection/mixin/features/bedrock/inventory/MixinInventoryScreen.java

@@ -23,6 +23,7 @@
 
 import com.viaversion.viafabricplus.ViaFabricPlusImpl;
 import com.viaversion.viafabricplus.protocoltranslator.ProtocolTranslator;
+import com.viaversion.viaversion.api.connection.UserConnection;
 import com.viaversion.viaversion.api.protocol.packet.PacketWrapper;
 import com.viaversion.viaversion.api.type.Types;
 import net.minecraft.client.gui.screens.inventory.InventoryScreen;
@@ -43,13 +44,10 @@ public class MixinInventoryScreen {
     @Inject(method = "<init>", at = @At("RETURN"))
     private void sendBedrockPacket(CallbackInfo ci) {
         if (ProtocolTranslator.getTargetVersion().equalTo(BedrockProtocolVersion.bedrockLatest)) {
-            EntityTracker entityTracker = ProtocolTranslator.getPlayNetworkUserConnection().get(EntityTracker.class);
-            if (entityTracker == null) {
-                ViaFabricPlusImpl.INSTANCE.getLogger().error("Could not get bedrock entity tracker");
-                return;
-            }
+            final UserConnection connection = ProtocolTranslator.getPlayNetworkUserConnection();
+            final EntityTracker entityTracker = connection.get(EntityTracker.class);
 
-            final PacketWrapper interact = PacketWrapper.create(ServerboundBedrockPackets.INTERACT, ProtocolTranslator.getPlayNetworkUserConnection());
+            final PacketWrapper interact = PacketWrapper.create(ServerboundBedrockPackets.INTERACT, connection);
             interact.write(Types.UNSIGNED_BYTE, (short) InteractPacket_Action.OpenInventory.getValue()); // action
             interact.write(BedrockTypes.UNSIGNED_VAR_LONG, entityTracker.getClientPlayer().runtimeId()); // target entity runtime id
             interact.write(BedrockTypes.OPTIONAL_POSITION_3F, null); // position