fix an error in set_entity_data

1 files changed, 7 insertions, 3 deletions

diff --git a/azalea-client/src/plugins/packet/game/mod.rs b/azalea-client/src/plugins/packet/game/mod.rs
index 5f3e125f..f3670c71 100644
--- a/azalea-client/src/plugins/packet/game/mod.rs
+++ b/azalea-client/src/plugins/packet/game/mod.rs
@@ -742,9 +742,13 @@ impl GamePacketHandler<'_> {
                 return;
             };
 
-            let entity_kind = *entity_kind_query
-                .get(entity)
-                .expect("EntityKind component should always be present for entities");
+            let Ok(entity_kind) = entity_kind_query.get(entity) else {
+                debug!(
+                    "Server sent an entity data packet for an entity id ({}) that we have indexed as {entity} but they don't have EntityKind. Maybe a second local client that just disconnected?",
+                    p.id
+                );
+                return;
+            };
 
             debug!("Got set entity data packet {p:?} for entity of kind {entity_kind:?}");