chore: use BlockSnapshot instead of Block in FlatteningRegion

ScreamingSandals · Aug 18, 2024 · 86fe777 · 86fe777
1 parent bec9f34
commit 86fe777
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/plugin/common/src/main/java/org/screamingsandals/bedwars/region/FlatteningRegion.java b/plugin/common/src/main/java/org/screamingsandals/bedwars/region/FlatteningRegion.java
@@ -33,7 +33,7 @@
 
 public class FlatteningRegion implements BWRegion {
     private final List<Location> builtBlocks = new ArrayList<>();
-    private final Map<Location, Block> brokenOriginalBlocks = new HashMap<>();
+    private final Map<Location, BlockSnapshot> brokenOriginalBlocks = new HashMap<>();
 
     @Override
     public boolean isLocationModifiedDuringGame(Location loc) {
@@ -42,7 +42,7 @@ public boolean isLocationModifiedDuringGame(Location loc) {
 
     @Override
     public void putOriginalBlock(Location loc, BlockSnapshot block) {
-        brokenOriginalBlocks.put(loc, block.block());
+        brokenOriginalBlocks.put(loc, block);
     }
 
     @Override
@@ -80,7 +80,7 @@ public void regen() {
             if (!chunk.isLoaded()) {
                 chunk.load();
             }
-            block.getKey().getBlock().block(block.getValue());
+            block.getValue().updateBlock(true, false);
         }
         brokenOriginalBlocks.clear();
     }