diff --git a/forge-ai/src/main/java/forge/ai/GameState.java b/forge-ai/src/main/java/forge/ai/GameState.java
index 7053c32ad9c..60f4a6c076c 100644
--- a/forge-ai/src/main/java/forge/ai/GameState.java
+++ b/forge-ai/src/main/java/forge/ai/GameState.java
@@ -639,6 +639,9 @@ protected void applyGameOnThread(final Game game) {
         // prevent interactions with objects from old state
         game.copyLastState();
 
+        // Store snapshot for restoring
+        game.stashGameState();
+
         // Set negative or zero life after state effects if need be, important for some puzzles that rely on
         // pre-setting negative life (e.g. PS_NEO4).
         for (int i = 0; i < playerStates.size(); i++) {