make compatible with any DragonEggBlock or subclass

1 files changed, 2 insertions, 2 deletions

diff --git a/src/main/java/wtf/kity/uncrackable/mixin/BlockBehaviourMixin.java b/src/main/java/wtf/kity/uncrackable/mixin/BlockBehaviourMixin.java
index 506b0c9..428d195 100644
--- a/src/main/java/wtf/kity/uncrackable/mixin/BlockBehaviourMixin.java
+++ b/src/main/java/wtf/kity/uncrackable/mixin/BlockBehaviourMixin.java
@@ -5,7 +5,7 @@ import com.llamalad7.mixinextras.injector.wrapoperation.Operation;
 import net.minecraft.core.BlockPos;
 import net.minecraft.world.entity.item.FallingBlockEntity;
 import net.minecraft.world.level.BlockGetter;
-import net.minecraft.world.level.block.Blocks;
+import net.minecraft.world.level.block.DragonEggBlock;
 import net.minecraft.world.level.block.state.BlockBehaviour;
 import net.minecraft.world.level.block.state.BlockState;
 import net.minecraft.world.level.material.PushReaction;
@@ -25,7 +25,7 @@ public class BlockBehaviourMixin {
         if (!shape.isEmpty()
                 && context instanceof EntityCollisionContext entityContext
                 && entityContext.getEntity() instanceof FallingBlockEntity fallingBlock
-                && fallingBlock.getBlockState().is(Blocks.DRAGON_EGG)
+                && fallingBlock.getBlockState().getBlock() instanceof DragonEggBlock
                 && state.getPistonPushReaction() != PushReaction.DESTROY
         ) {
             return Shapes.block();