Fix bad affix list picking

1 files changed, 4 insertions, 0 deletions

diff --git a/src/main/java/darkknight/jewelrycraft/affixes/AffixMods.java b/src/main/java/darkknight/jewelrycraft/affixes/AffixMods.java
index 266d70d..701b00f 100644
--- a/src/main/java/darkknight/jewelrycraft/affixes/AffixMods.java
+++ b/src/main/java/darkknight/jewelrycraft/affixes/AffixMods.java
@@ -139,6 +139,8 @@ public class AffixMods {
 			}
 		}
 
+		if (pickList.size() <= 0) pickList = generalPrefixGen;
+		
 		WeightedRandomAffix wra = ((WeightedRandomAffix) WeightedRandom.getRandomItem(random, pickList));
 
 		return wra.getAffix(random);
@@ -167,6 +169,8 @@ public class AffixMods {
 			}
 		}
 
+		if (pickList.size() <= 0) pickList = generalSuffixGen;
+		
 		WeightedRandomAffix wra = ((WeightedRandomAffix) WeightedRandom.getRandomItem(random, pickList));
 
 		return wra.getAffix(random);