From e626370cff646b27303c67454757dc104a41c8fa Mon Sep 17 00:00:00 2001
From: NichtStudioCode <51272202+NichtStudioCode@users.noreply.github.com>
Date: Tue, 20 Aug 2024 17:26:47 +0200
Subject: [PATCH] Return new FakeItemDisplay on MultiModel#add

---
 .../main/kotlin/xyz/xenondevs/nova/world/model/MultiModel.kt   | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/nova/src/main/kotlin/xyz/xenondevs/nova/world/model/MultiModel.kt b/nova/src/main/kotlin/xyz/xenondevs/nova/world/model/MultiModel.kt
index 0924a928bff..62c964efea6 100644
--- a/nova/src/main/kotlin/xyz/xenondevs/nova/world/model/MultiModel.kt
+++ b/nova/src/main/kotlin/xyz/xenondevs/nova/world/model/MultiModel.kt
@@ -17,9 +17,10 @@ abstract class MultiModel {
         models.forEach(this::add)
     }
     
-    fun add(model: Model) {
+    fun add(model: Model): FakeItemDisplay {
         val itemDisplay = model.createFakeItemDisplay()
         models.put(model, itemDisplay)?.remove()
+        return itemDisplay
     }
     
     fun clear() {