From 39274f341ea2818be0f00a6d2dd73e856677de5e Mon Sep 17 00:00:00 2001
From: IThundxr <me@ithundxr.dev>
Date: Sat, 12 Oct 2024 17:33:37 -0400
Subject: [PATCH] fix

---
 .../dev/engine_room/flywheel/impl/compat/EmbeddiumCompat.java | 4 +++-
 1 file changed, 3 insertions(+), 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
index ba99200ec..c7a8834d3 100644
--- 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);
 		});
 	}
 }