diff --git a/common/src/main/java/com/dfsek/terra/api/util/collections/ProbabilityCollection.java b/common/src/main/java/com/dfsek/terra/api/util/collections/ProbabilityCollection.java index 1a49948db..77ded4d59 100644 --- a/common/src/main/java/com/dfsek/terra/api/util/collections/ProbabilityCollection.java +++ b/common/src/main/java/com/dfsek/terra/api/util/collections/ProbabilityCollection.java @@ -23,7 +23,7 @@ public class ProbabilityCollection implements Collection { public ProbabilityCollection add(E item, int probability) { if(!cont.containsKey(item)) size++; - cont.computeIfAbsent(item, i -> new MutableInteger(0)).increment(); + cont.computeIfAbsent(item, i -> new MutableInteger(0)).add(probability); int oldLength = array.length; Object[] newArray = new Object[array.length + probability]; System.arraycopy(array, 0, newArray, 0, array.length); // Expand array.