From 23fc19430acfd75f0f92d37680cdb2017b477ab3 Mon Sep 17 00:00:00 2001
From: themode <themode@outlook.fr>
Date: Wed, 30 Dec 2020 20:29:46 +0100
Subject: [PATCH] Fix entity uuid map memory leak

---
 src/main/java/net/minestom/server/entity/Entity.java | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/main/java/net/minestom/server/entity/Entity.java b/src/main/java/net/minestom/server/entity/Entity.java
index bb8b16cc8..f01ffe1a0 100644
--- a/src/main/java/net/minestom/server/entity/Entity.java
+++ b/src/main/java/net/minestom/server/entity/Entity.java
@@ -1219,12 +1219,13 @@ public abstract class Entity implements Viewable, EventHandler, DataContainer, P
     /**
      * Removes the entity from the server immediately.
      * <p>
-     * WARNING: this do not trigger the {@link EntityDeathEvent} event.
+     * WARNING: this does not trigger {@link EntityDeathEvent}.
      */
     public void remove() {
         this.removed = true;
         this.shouldRemove = true;
-        entityById.remove(id);
+        Entity.entityById.remove(id);
+        Entity.entityByUuid.remove(uuid);
         if (instance != null)
             instance.UNSAFE_removeEntity(this);
     }