Now not displaying entity when it is unloaded

Fixes ToroCraft#82
CTAK-CO6AK · Mar 4, 2021 · 90a4612 · 90a4612
1 parent 0e3e1a1
commit 90a4612
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/main/java/net/torocraft/torohealth/display/Hud.java b/src/main/java/net/torocraft/torohealth/display/Hud.java
@@ -99,7 +99,12 @@ private void draw(MatrixStack matrix, float x, float y, float scale) {
     if (entity == null) {
       return;
     }
-
+
+    if(client.world.getEntityById(entity.getEntityId()) != entity) {
+      setEntityWork(null);
+      return;
+    }
+
     matrix.push();
     matrix.scale(scale, scale, scale);
     matrix.translate(x - 10, y - 10, 0);