minor fix to fire fix

DynamicTreesTeam · Sep 8, 2024 · f7006f7 · f7006f7
1 parent 2992aa6
commit f7006f7
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/src/main/java/com/ferreusveritas/dynamictrees/block/branch/BranchBlock.java b/src/main/java/com/ferreusveritas/dynamictrees/block/branch/BranchBlock.java
@@ -577,12 +577,14 @@ public void onRemove(BlockState state, Level level, BlockPos pos, BlockState new
             return;
         }
 
-        boolean foundFire = false;
-        for (Direction offset : Direction.values()){
-            BlockPos offPos = pos.offset(offset.getNormal());
-            if (level.getBlockState(offPos).is(BlockTags.FIRE)){
-                foundFire = true;
-                break;
+        boolean foundFire = toBlockState.is(BlockTags.FIRE);
+        if (!foundFire){
+            for (Direction offset : Direction.values()){
+                BlockPos offPos = pos.offset(offset.getNormal());
+                if (level.getBlockState(offPos).is(BlockTags.FIRE)){
+                    foundFire = true;
+                    break;
+                }
             }
         }