diff --git a/src/main/java/org/dimdev/dimdoors/pockets/generator/PocketGenerator.java b/src/main/java/org/dimdev/dimdoors/pockets/generator/PocketGenerator.java index 8a2bafde..130ed60c 100644 --- a/src/main/java/org/dimdev/dimdoors/pockets/generator/PocketGenerator.java +++ b/src/main/java/org/dimdev/dimdoors/pockets/generator/PocketGenerator.java @@ -41,6 +41,7 @@ public abstract class PocketGenerator implements Weighted tags = new ArrayList<>(); @@ -83,6 +84,8 @@ public abstract class PocketGenerator implements Weighted())); } + public Boolean getSetupLoot() { + return setupLoot; + } + public void applyModifiers(PocketGenerationParameters parameters, RiftManager manager) { for (Modifier modifier : modifierList) { modifier.apply(parameters, manager); diff --git a/src/main/java/org/dimdev/dimdoors/pockets/virtual/reference/PocketGeneratorReference.java b/src/main/java/org/dimdev/dimdoors/pockets/virtual/reference/PocketGeneratorReference.java index 62d761cb..047a6132 100644 --- a/src/main/java/org/dimdev/dimdoors/pockets/virtual/reference/PocketGeneratorReference.java +++ b/src/main/java/org/dimdev/dimdoors/pockets/virtual/reference/PocketGeneratorReference.java @@ -99,7 +99,7 @@ public abstract class PocketGeneratorReference extends VirtualSingularPocket { generator.applyModifiers(parameters, manager); this.applyModifiers(parameters, manager); - generator.setup(pocket, parameters, setupLoot != null ? setupLoot : true); + generator.setup(pocket, parameters, setupLoot != null ? setupLoot : (generator.getSetupLoot()) != null ? generator.getSetupLoot() : true); return pocket; }