fix

Engine-Room · Oct 12, 2024 · 39274f3 · 39274f3
1 parent dd80ea4
commit 39274f3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/common/src/main/java/dev/engine_room/flywheel/impl/compat/EmbeddiumCompat.java b/common/src/main/java/dev/engine_room/flywheel/impl/compat/EmbeddiumCompat.java
@@ -2,10 +2,12 @@
 
 import org.embeddedt.embeddium.api.ChunkDataBuiltEvent;
 
+import dev.engine_room.flywheel.lib.visualization.VisualizationHelper;
+
 public class EmbeddiumCompat {
 	public static void init() {
 		ChunkDataBuiltEvent.BUS.addListener(event -> {
-			event.getDataBuilder().removeBlockEntitiesIf(InstancedRenderDispatcher::tryAddBlockEntity);
+			event.getDataBuilder().removeBlockEntitiesIf(VisualizationHelper::tryAddBlockEntity);
 		});
 	}
 }