From 9beaf7119641c8b50036ebc83a30c6c5f18296df Mon Sep 17 00:00:00 2001
From: Nassim Jahnke <nassim@njahnke.dev>
Date: Fri, 22 Mar 2024 20:39:31 +0100
Subject: [PATCH] Handle new argument types

---
 .../Protocol1_20_3To1_20_5.java                     | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/common/src/main/java/com/viaversion/viabackwards/protocol/protocol1_20_3to1_20_5/Protocol1_20_3To1_20_5.java b/common/src/main/java/com/viaversion/viabackwards/protocol/protocol1_20_3to1_20_5/Protocol1_20_3To1_20_5.java
index 6617f253..592f66f5 100644
--- a/common/src/main/java/com/viaversion/viabackwards/protocol/protocol1_20_3to1_20_5/Protocol1_20_3To1_20_5.java
+++ b/common/src/main/java/com/viaversion/viabackwards/protocol/protocol1_20_3to1_20_5/Protocol1_20_3To1_20_5.java
@@ -139,7 +139,18 @@ public final class Protocol1_20_3To1_20_5 extends BackwardsProtocol<ClientboundP
             response.sendToServer(Protocol1_20_3To1_20_5.class);
         });
 
-        new CommandRewriter1_19_4<>(this).registerDeclareCommands1_19(ClientboundPackets1_20_5.DECLARE_COMMANDS);
+        new CommandRewriter1_19_4<ClientboundPacket1_20_5>(this) {
+            @Override
+            public void handleArgument(final PacketWrapper wrapper, final String argumentType) throws Exception {
+                if (argumentType.equals("minecraft:loot_table")
+                    || argumentType.equals("minecraft:loot_predicate")
+                    || argumentType.equals("minecraft:loot_modifier")) {
+                    wrapper.write(Type.VAR_INT, 0);
+                } else {
+                    super.handleArgument(wrapper, argumentType);
+                }
+            }
+        }.registerDeclareCommands1_19(ClientboundPackets1_20_5.DECLARE_COMMANDS);
 
         cancelClientbound(ClientboundPackets1_20_5.DEBUG_SAMPLE);
     }