Fix #987

FlansMods · Jan 7, 2019 · 8667bbd · 8667bbd
1 parent 71dc31d
commit 8667bbd
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/src/main/java/com/flansmod/apocalypse/common/FlansModLootGenerator.java b/src/main/java/com/flansmod/apocalypse/common/FlansModLootGenerator.java
@@ -6,6 +6,7 @@
 
 import net.minecraft.block.Block;
 import net.minecraft.entity.EntityLivingBase;
+import net.minecraft.init.Blocks;
 import net.minecraft.init.Items;
 import net.minecraft.inventory.EntityEquipmentSlot;
 import net.minecraft.item.ItemStack;
@@ -279,13 +280,21 @@ public Block getRandomWeaponBox(Random rand)
 		if(rand.nextInt(4) == 0)
 		{
 			//Get armour box
-			return ArmourBoxType.boxes.get(new ArrayList<>(ArmourBoxType.boxes.keySet()).get(rand.nextInt(ArmourBoxType.boxes.size()))).block;
+			if(ArmourBoxType.boxes.size() > 0)
+			{
+				return ArmourBoxType.boxes.get(new ArrayList<>(ArmourBoxType.boxes.keySet()).get(rand.nextInt(ArmourBoxType.boxes.size()))).block;
+			}
 		}
 		else
 		{
 			//Get weapon box
-			return GunBoxType.gunBoxMap.get(new ArrayList<>(GunBoxType.gunBoxMap.keySet()).get(rand.nextInt(GunBoxType.gunBoxMap.size()))).block;
+			if(GunBoxType.gunBoxMap.size() > 0)
+			{
+				return GunBoxType.gunBoxMap.get(new ArrayList<>(GunBoxType.gunBoxMap.keySet()).get(rand.nextInt(GunBoxType.gunBoxMap.size()))).block;
+			}
 		}
+
+		return Blocks.AIR;
 	}
 
 	public DriveableType getRandomDriveable(Random rand)