1 files changed, 8 insertions, 3 deletions
diff --git a/src/main/kotlin/moe/nea/firmament/events/BakeExtraModelsEvent.kt b/src/main/kotlin/moe/nea/firmament/events/BakeExtraModelsEvent.kt
index db073e3..f75bedc 100644
--- a/src/main/kotlin/moe/nea/firmament/events/BakeExtraModelsEvent.kt
+++ b/src/main/kotlin/moe/nea/firmament/events/BakeExtraModelsEvent.kt
@@ -5,11 +5,16 @@ import java.util.function.Consumer
 import net.minecraft.client.util.ModelIdentifier
 
 class BakeExtraModelsEvent(
-    private val addModel: Consumer<ModelIdentifier>,
+    private val addItemModel: Consumer<ModelIdentifier>,
+    private val addAnyModel: Consumer<ModelIdentifier>,
 ) : FirmamentEvent() {
 
-    fun addModel(modelIdentifier: ModelIdentifier) {
-        this.addModel.accept(modelIdentifier)
+    fun addNonItemModel(modelIdentifier: ModelIdentifier) {
+        this.addAnyModel.accept(modelIdentifier)
+    }
+
+    fun addItemModel(modelIdentifier: ModelIdentifier) {
+        this.addItemModel.accept(modelIdentifier)
     }
 
     companion object : FirmamentEventBus<BakeExtraModelsEvent>()