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 f1fa6c4..53c42a6 100644
--- a/src/main/java/pw/kaboom/extras/modules/entity/EntitySpawn.java
+++ b/src/main/java/pw/kaboom/extras/modules/entity/EntitySpawn.java
@@ -51,7 +51,7 @@ private void applyEntityChanges(final Entity entity) {
     }
 
     private boolean checkShouldRemoveEntities(final World world) {
-        final int worldEntityCount = world.getEntities().size();
+        final int worldEntityCount = world.getEntityCount();
 
         if (worldEntityCount > MAX_ENTITIES_PER_WORLD) {
             for (Entity entity : world.getEntities()) {