fix #1224 (#1256)

Potion-Studios · Oct 13, 2023 · f33d0a8 · f33d0a8
1 parent cf72c80
commit f33d0a8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Common/src/main/java/potionstudios/byg/common/blockentity/HypogealImperiumBlockEntity.java b/Common/src/main/java/potionstudios/byg/common/blockentity/HypogealImperiumBlockEntity.java
@@ -125,7 +125,8 @@ public static void playSound(Level level, BlockPos blockPos, BlockState blockSta
     }
 
     public static void setLit(Level level, BlockPos blockPos, BlockState blockState, HypogealImperiumBlockEntity hypogealImperiumBlockEntity) {
-        if (hypogealImperiumBlockEntity.isLit()) {
+        BlockState state = hypogealImperiumBlockEntity.getBlockState();
+        if (hypogealImperiumBlockEntity.isLit() && state.getBlock() instanceof HypogealImperiumBlock) {
             level.setBlock(hypogealImperiumBlockEntity.worldPosition, level.getBlockState(hypogealImperiumBlockEntity.worldPosition).setValue(HypogealImperiumBlock.LIT, hypogealImperiumBlockEntity.isLit()), 3);
             hypogealImperiumBlockEntity.setChanged();
         }