diff --git a/src/main/java/com/jozufozu/flywheel/FlywheelClient.java b/src/main/java/com/jozufozu/flywheel/FlywheelClient.java
index 068f6ea31..f38ca7b6b 100644
--- a/src/main/java/com/jozufozu/flywheel/FlywheelClient.java
+++ b/src/main/java/com/jozufozu/flywheel/FlywheelClient.java
@@ -5,6 +5,8 @@ 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 net.minecraftforge.eventbus.api.IEventBus;
 import net.minecraftforge.fml.javafmlmod.FMLJavaModLoadingContext;
 
@@ -20,5 +22,7 @@ public class FlywheelClient {
 
 		modEventBus.addListener(Contexts::flwInit);
 		modEventBus.addListener(Materials::flwInit);
+		modEventBus.addListener(PartialModel::onModelRegistry);
+		modEventBus.addListener(PartialModel::onModelBake);
 	}
 }