diff --git a/common/src/main/java/dev/latvian/mods/kubejs/block/state/BlockStatePredicate.java b/common/src/main/java/dev/latvian/mods/kubejs/block/state/BlockStatePredicate.java
index 11c0e50fa..80f38f81e 100644
--- a/common/src/main/java/dev/latvian/mods/kubejs/block/state/BlockStatePredicate.java
+++ b/common/src/main/java/dev/latvian/mods/kubejs/block/state/BlockStatePredicate.java
@@ -327,10 +327,10 @@ final class RegexMatch implements BlockStatePredicate {
 		public RegexMatch(Pattern p) {
 			pattern = p;
 			matchedBlocks = new LinkedHashSet<>();
-			for (var state : UtilsJS.getAllBlockStates()) {
-				var block = state.getBlock();
+			for (var entry : RegistryInfo.BLOCK.entrySet()) {
+				var block = entry.getValue();
 				if (!matchedBlocks.contains(block) && pattern.matcher(RegistryInfo.BLOCK.getId(block).toString()).find()) {
-					matchedBlocks.add(state.getBlock());
+					matchedBlocks.add(block);
 				}
 			}
 		}