Fix error on spawn when categorizing entities not registered with geary

MineInAbyss · Nov 26, 2023 · 26fa305 · 26fa305
1 parent 0a7290f
commit 26fa305
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mobzy-spawning/src/main/kotlin/com/mineinabyss/mobzy/spawning/vertical/SpawnInfo.kt b/mobzy-spawning/src/main/kotlin/com/mineinabyss/mobzy/spawning/vertical/SpawnInfo.kt
@@ -84,7 +84,7 @@ class SpawnInfo(
     companion object {
         //TODO perhaps give normal mobs prefab keys too to make this more type safe
         fun categorizeByType(mobs: Collection<Entity>): Map<PrefabKey?, Int> =
-            mobs.groupingBy { it.toGeary().prefabs.first().get<PrefabKey>() }.eachCount()
+            mobs.groupingBy { it.toGeary().prefabs.firstOrNull()?.get<PrefabKey>() }.eachCount()
     }
 }