diff --git a/src/magic/headless/HeadlessGameController.java b/src/magic/headless/HeadlessGameController.java
index f8f34ee09d..f8dc408b93 100644
--- a/src/magic/headless/HeadlessGameController.java
+++ b/src/magic/headless/HeadlessGameController.java
@@ -46,7 +46,7 @@ public class HeadlessGameController implements IGameController {
     private Object[] getAIChoiceResults(final MagicEvent event) {
         //dynamically get the AI based on the player's index
         final MagicPlayer player = event.getPlayer();
-        final MagicAI ai = game.getDuel().getAIs()[player.getIndex()];
+        final MagicAI ai = player.getPlayerDefinition().getPlayerProfile().getAiType().getAI();
         return ai.findNextEventChoiceResults(game, player);
     }
 }
diff --git a/src/magic/ui/SwingGameController.java b/src/magic/ui/SwingGameController.java
index 7c839af47a..898978937e 100644
--- a/src/magic/ui/SwingGameController.java
+++ b/src/magic/ui/SwingGameController.java
@@ -587,7 +587,7 @@ public class SwingGameController implements IUIGameController, ILogBookListener
 
         //dynamically get the AI based on the player's index
         final MagicPlayer player = event.getPlayer();
-        final MagicAI ai = game.getDuel().getAIs()[player.getIndex()];
+        final MagicAI ai = player.getPlayerDefinition().getPlayerProfile().getAiType().getAI();
         return ai.findNextEventChoiceResults(game, player);
     }