From b5e1aa0ac093ddab3eaa93d5576532ab08d94540 Mon Sep 17 00:00:00 2001
From: Luna <lunahatesgogle@gmail.com>
Date: Thu, 17 Aug 2023 18:49:17 -0300
Subject: [PATCH] Don't randomly cancel TNT unless limit reached

---
 .../java/pw/kaboom/extras/modules/entity/EntitySpawn.java     | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/main/java/pw/kaboom/extras/modules/entity/EntitySpawn.java b/src/main/java/pw/kaboom/extras/modules/entity/EntitySpawn.java
index bbf3260..43cb8b9 100644
--- a/src/main/java/pw/kaboom/extras/modules/entity/EntitySpawn.java
+++ b/src/main/java/pw/kaboom/extras/modules/entity/EntitySpawn.java
@@ -240,7 +240,9 @@ public final class EntitySpawn implements Listener {
 
     @EventHandler
     void onTNTPrime(final TNTPrimeEvent event) {
-        if (ThreadLocalRandom.current().nextBoolean()) {
+        if (event.getBlock()
+                .getWorld().getEntitiesByClass(TNTPrimed.class).size() >= MAX_TNTS_PER_WORLD
+                && ThreadLocalRandom.current().nextBoolean()) {
             event.setCancelled(true);
         }
     }