diff --git a/src/main/java/com/jozufozu/flywheel/FlywheelClient.java b/src/main/java/com/jozufozu/flywheel/FlywheelClient.java
index 98ecf0473..9cf2e6f79 100644
--- a/src/main/java/com/jozufozu/flywheel/FlywheelClient.java
+++ b/src/main/java/com/jozufozu/flywheel/FlywheelClient.java
@@ -4,7 +4,7 @@ import com.jozufozu.flywheel.backend.Backend;
 import com.jozufozu.flywheel.core.AtlasStitcher;
 import com.jozufozu.flywheel.core.Contexts;
 import com.jozufozu.flywheel.core.Materials;
-
+import com.jozufozu.flywheel.core.PartialModel;
 import com.jozufozu.flywheel.vanilla.VanillaInstances;
 
 import net.minecraftforge.eventbus.api.IEventBus;
@@ -22,6 +22,8 @@ public class FlywheelClient {
 
 		modEventBus.addListener(Contexts::flwInit);
 		modEventBus.addListener(Materials::flwInit);
+		modEventBus.addListener(PartialModel::onModelRegistry);
+		modEventBus.addListener(PartialModel::onModelBake);
 
 		VanillaInstances.init();
 	}