From 90a461272845451bb0188b7d4bda30a64ce8cc6a Mon Sep 17 00:00:00 2001
From: CTAK-CO6AK <72935702+CTAK-CO6AK@users.noreply.github.com>
Date: Thu, 4 Mar 2021 14:08:26 +0400
Subject: [PATCH 1/2] Now not displaying entity when it is unloaded

Fixes #82
---
 src/main/java/net/torocraft/torohealth/display/Hud.java | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/main/java/net/torocraft/torohealth/display/Hud.java b/src/main/java/net/torocraft/torohealth/display/Hud.java
index 559e3e3a..37043697 100644
--- 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);

From 68921a0f5c3937097a42085201906a89717aaf05 Mon Sep 17 00:00:00 2001
From: CTAK-CO6AK <72935702+CTAK-CO6AK@users.noreply.github.com>
Date: Thu, 4 Mar 2021 14:09:47 +0400
Subject: [PATCH 2/2] Reset BarState when entity is unloaded

Fixes #82
---
 src/main/java/net/torocraft/torohealth/bars/BarStates.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main/java/net/torocraft/torohealth/bars/BarStates.java b/src/main/java/net/torocraft/torohealth/bars/BarStates.java
index 17e282d1..7a446ed5 100644
--- a/src/main/java/net/torocraft/torohealth/bars/BarStates.java
+++ b/src/main/java/net/torocraft/torohealth/bars/BarStates.java
@@ -17,7 +17,7 @@ public class BarStates {
   public static BarState getState(LivingEntity entity) {
     int id = entity.getEntityId();
     BarState state = STATES.get(id);
-    if (state == null) {
+    if (state == null || state.entity != entity) {
       state = new BarState(entity);
       STATES.put(id, state);
     }