From 8c0032261275174c83cd90320bd7c99e2d41b1cc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tam=C3=A1s=20=C3=81d=C3=A1m?= <atomi2500@gmail.com>
Date: Mon, 12 Dec 2022 19:19:44 +0200
Subject: [PATCH] make ButtonContainer a bit more responsive

---
 src/main/java/com/vamk/tbg/ui/ButtonContainer.java | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/main/java/com/vamk/tbg/ui/ButtonContainer.java b/src/main/java/com/vamk/tbg/ui/ButtonContainer.java
index a648fec..e711af5 100644
--- a/src/main/java/com/vamk/tbg/ui/ButtonContainer.java
+++ b/src/main/java/com/vamk/tbg/ui/ButtonContainer.java
@@ -112,6 +112,9 @@ private void onEntityPlays(EntityPlaysSignal signal) {
 
             button.setEnabled(userControlled);
         }
+
+        // Enable all entity buttons again
+        if (userControlled) this.entityButtons.values().forEach(x -> x.setEnabled(true));
     }
 
     /**