1 files changed, 5 insertions, 2 deletions
diff --git a/src/texturePacks/java/moe/nea/firmament/features/texturepack/FirmamentModelPredicate.kt b/src/texturePacks/java/moe/nea/firmament/features/texturepack/FirmamentModelPredicate.kt
index d11fec0..e020d66 100644
--- a/src/texturePacks/java/moe/nea/firmament/features/texturepack/FirmamentModelPredicate.kt
+++ b/src/texturePacks/java/moe/nea/firmament/features/texturepack/FirmamentModelPredicate.kt
@@ -1,8 +1,11 @@
-
 package moe.nea.firmament.features.texturepack
 
+import kotlinx.serialization.Serializable
+import net.minecraft.entity.LivingEntity
 import net.minecraft.item.ItemStack
 
+@Serializable(with = FirmamentRootPredicateSerializer::class)
 interface FirmamentModelPredicate {
-    fun test(stack: ItemStack): Boolean
+	fun test(stack: ItemStack, holder: LivingEntity?): Boolean = test(stack)
+	fun test(stack: ItemStack): Boolean = test(stack, null)
 }