Fix pickblock for quick armor stands

NucleoidMC · Dec 14, 2024 · a81750f · a81750f
1 parent 62f29a2
commit a81750f
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/main/java/xyz/nucleoid/extras/lobby/entity/QuickArmorStandEntity.java b/src/main/java/xyz/nucleoid/extras/lobby/entity/QuickArmorStandEntity.java
@@ -4,11 +4,13 @@
 import net.minecraft.entity.EntityType;
 import net.minecraft.entity.damage.DamageSource;
 import net.minecraft.entity.decoration.ArmorStandEntity;
+import net.minecraft.item.ItemStack;
 import net.minecraft.server.network.ServerPlayerEntity;
 import net.minecraft.server.world.ServerWorld;
 import net.minecraft.util.math.Vec3d;
 import net.minecraft.world.World;
 import xyz.nucleoid.extras.lobby.NEEntities;
+import xyz.nucleoid.extras.lobby.NEItems;
 import xyz.nucleoid.packettweaker.PacketContext;
 
 public class QuickArmorStandEntity extends ArmorStandEntity implements PolymerEntity {
@@ -34,6 +36,11 @@ public boolean isInvulnerableTo(ServerWorld world, DamageSource damageSource) {
         return super.isInvulnerableTo(world, damageSource);
     }
 
+    @Override
+    public ItemStack getPickBlockStack() {
+        return NEItems.QUICK_ARMOR_STAND.getDefaultStack();
+    }
+
     @Override
     protected void tickCramming() {