From b2aeaff2d6d9e15b92df1eeab99086ba0d68f66e Mon Sep 17 00:00:00 2001
From: Hanmac <hanmac@cardforge.org>
Date: Thu, 18 Aug 2016 15:48:23 +0000
Subject: [PATCH] ComputerUtil: add more LastState

---
 forge-ai/src/main/java/forge/ai/ComputerUtil.java | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/forge-ai/src/main/java/forge/ai/ComputerUtil.java b/forge-ai/src/main/java/forge/ai/ComputerUtil.java
index d5e7b91d83d..4f83dd57cdd 100644
--- a/forge-ai/src/main/java/forge/ai/ComputerUtil.java
+++ b/forge-ai/src/main/java/forge/ai/ComputerUtil.java
@@ -210,6 +210,8 @@ public class ComputerUtil {
         final Card source = sa.getHostCard();
         if (sa.isSpell() && !source.isCopiedSpell()) {
             source.setCastSA(sa);
+            sa.setLastStateBattlefield(game.getLastStateBattlefield());
+            sa.setLastStateGraveyard(game.getLastStateGraveyard());
             sa.setHostCard(game.getAction().moveToStack(source));
         }
         final Cost cost = sa.getPayCosts();
@@ -231,6 +233,8 @@ public class ComputerUtil {
         final Card source = sa.getHostCard();
         if (sa.isSpell() && !source.isCopiedSpell()) {
             source.setCastSA(sa);
+            sa.setLastStateBattlefield(game.getLastStateBattlefield());
+            sa.setLastStateGraveyard(game.getLastStateGraveyard());
             sa.setHostCard(ai.getGame().getAction().moveToStack(source));
         }
 
@@ -248,6 +252,8 @@ public class ComputerUtil {
         final Card source = newSA.getHostCard();
         if (newSA.isSpell() && !source.isCopiedSpell()) {
             source.setCastSA(newSA);
+            sa.setLastStateBattlefield(game.getLastStateBattlefield());
+            sa.setLastStateGraveyard(game.getLastStateGraveyard());
             newSA.setHostCard(game.getAction().moveToStack(source));
         }
 
@@ -265,6 +271,8 @@ public class ComputerUtil {
             final Card source = sa.getHostCard();
             if (sa.isSpell() && !source.isCopiedSpell()) {
                 source.setCastSA(sa);
+                sa.setLastStateBattlefield(game.getLastStateBattlefield());
+                sa.setLastStateGraveyard(game.getLastStateGraveyard());
                 sa.setHostCard(game.getAction().moveToStack(source));
             }