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);
 		});
 	}
 }