fix a check HEAD main

1 files changed, 4 insertions, 2 deletions

diff --git a/src/main/java/wtf/kity/uncrackable/mixin/FallingBlockEntityMixin.java b/src/main/java/wtf/kity/uncrackable/mixin/FallingBlockEntityMixin.java
index 3d76afb..bd45488 100644
--- a/src/main/java/wtf/kity/uncrackable/mixin/FallingBlockEntityMixin.java
+++ b/src/main/java/wtf/kity/uncrackable/mixin/FallingBlockEntityMixin.java
@@ -15,6 +15,7 @@ import net.minecraft.world.level.block.DragonEggBlock;
 import net.minecraft.world.level.block.entity.BlockEntity;
 import net.minecraft.world.level.block.state.BlockState;
 import net.minecraft.world.phys.Vec3;
+import org.jetbrains.annotations.Nullable;
 import org.objectweb.asm.Opcodes;
 import org.spongepowered.asm.mixin.Mixin;
 import org.spongepowered.asm.mixin.injection.At;
@@ -83,14 +84,15 @@ public abstract class FallingBlockEntityMixin extends Entity {
             BlockState instance,
             BlockPlaceContext blockPlaceContext,
             Operation<Boolean> original,
+            @Local Block block,
             @Local BlockPos blockPos
     ) {
         if (original.call(instance, blockPlaceContext)) {
             return true;
         }
 
-        if (instance.getBlock() instanceof DragonEggBlock) {
-            BlockEntity blockentity = instance.hasBlockEntity() ? this.level().getBlockEntity(blockPos) : null;
+        if (block instanceof DragonEggBlock) {
+            @Nullable BlockEntity blockentity = this.level().getBlockEntity(blockPos);
             Block.dropResources(instance, this.level(), blockPos, blockentity);
             instance.onDestroyedByPushReaction(this.level(), blockPos, Direction.DOWN, this.level().getFluidState(blockPos));
             return true;