feat(BlockStateSafeGetter): use block type's default state

AllayMC · Oct 15, 2024 · b52f8f6 · b52f8f6
1 parent 1cb3bb6
commit b52f8f6
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/api/src/main/java/org/allaymc/api/block/type/BlockStateSafeGetter.java b/api/src/main/java/org/allaymc/api/block/type/BlockStateSafeGetter.java
@@ -93,9 +93,11 @@ public BlockState blockState() {
             // Add missing properties
             var updatedStates = updatedNbt.getCompound("states");
             var updatedStatesBuilder = updatedStates.toBuilder();
-            for (var entry : blockType.getProperties().entrySet()) {
-                if (updatedStatesBuilder.containsKey(entry.getKey())) continue;
-                updatedStatesBuilder.put(entry.getKey(), entry.getValue().createDefaultValue().getSerializedValue());
+            for (var entry : blockType.getDefaultState().getPropertyValues().entrySet()) {
+                if (updatedStatesBuilder.containsKey(entry.getKey().getName())) {
+                    continue;
+                }
+                updatedStatesBuilder.put(entry.getKey().getName(), entry.getValue().getSerializedValue());
             }
             updatedStates = updatedStatesBuilder.build();