Update entities.js

aesthetic0001 · Dec 22, 2023 · 066e104 · 066e104
1 parent 759372c
commit 066e104
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/lib/plugins/entities.js b/lib/plugins/entities.js
@@ -349,15 +349,18 @@ function inject (bot) {
             entity.heldItem = entity.equipment[0] // Update held item like prismarine-entity does upon equipment updates
         }
 
-        if (eventName === 'entityDead' && entity.passengers) {
-            entity.passengers.forEach((passengerId) => {
-                const passengerEntity = fetchEntity(passengerId)
-                if (!passengerEntity) return
-                passengerEntity.vehicle = null
-            })
+        if (eventName === 'entityDead') {
+            if (entity.passengers) {
+                entity.passengers.forEach((passengerId) => {
+                    const passengerEntity = fetchEntity(passengerId)
+                    if (!passengerEntity) return
+                    passengerEntity.vehicle = null
+                })
+            }
         }
 
         if (eventName) bot.emit(eventName, entity)
+        if (eventName === 'entityDead') delete bot.entities[packet.entityId]
     })
 
     bot._client.on('attach_entity', (packet) => {