Fix resource gen

TheEpicBlock · Aug 13, 2022 · e2002c7 · e2002c7
1 parent 27efb50
commit e2002c7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/java/nl/theepicblock/polyconfig/mixin/OnBuildPolyMap.java b/src/main/java/nl/theepicblock/polyconfig/mixin/OnBuildPolyMap.java
@@ -28,8 +28,9 @@ public void recheckPolys(CallbackInfoReturnable<PolyMap> cir) {
             if (poly instanceof CustomBlockPoly customPoly) {
                 newBlockPolys.put(block, new FunctionBlockStatePoly(
                         block,
-                        (state, registry) -> {
+                        (state, isUniqueCallback) -> {
                             var oldClientState = customPoly.getClientBlock(state);
+                            isUniqueCallback.set(!customPoly.toBeRechecked.containsKey(state));
                             if (customPoly.toBeRechecked.containsKey(state) && this.blockPolys.containsKey(oldClientState.getBlock())) {
                                 // Recheck the client state
                                 return this.blockPolys.get(oldClientState.getBlock()).getClientBlock(state);