diff --git a/forge-gui-mobile/src/forge/screens/quest/NewQuestScreen.java b/forge-gui-mobile/src/forge/screens/quest/NewQuestScreen.java index 51f14a71410..405e10c00de 100644 --- a/forge-gui-mobile/src/forge/screens/quest/NewQuestScreen.java +++ b/forge-gui-mobile/src/forge/screens/quest/NewQuestScreen.java @@ -1,10 +1,6 @@ package forge.screens.quest; -import java.util.ArrayList; -import java.util.HashSet; -import java.util.List; -import java.util.Map.Entry; -import java.util.Set; +import java.util.*; import com.badlogic.gdx.utils.Align; @@ -549,13 +545,8 @@ public class NewQuestScreen extends FScreen { fmtPrizes = fmtStartPool; if (null == fmtPrizes && dckStartPool != null) { // build it form deck Set sets = new HashSet<>(); - for (Entry c : dckStartPool.getMain()) { - sets.add(c.getKey().getEdition()); - } - if (dckStartPool.has(DeckSection.Sideboard)) { - for (Entry c : dckStartPool.get(DeckSection.Sideboard)) { - sets.add(c.getKey().getEdition()); - } + for(PaperCard card : dckStartPool.getAllCardsInASinglePool().toFlatList()) { + sets.add(card.getEdition()); } fmtPrizes = new GameFormat("From deck", sets, null); }