diff --git a/Mage.Tests/src/test/java/org/mage/test/cards/triggers/dies/ReturnOnlyFromGraveyardTest.java b/Mage.Tests/src/test/java/org/mage/test/cards/triggers/dies/ReturnOnlyFromGraveyardTest.java
index cc8bb46cf7..fba692a07d 100644
--- a/Mage.Tests/src/test/java/org/mage/test/cards/triggers/dies/ReturnOnlyFromGraveyardTest.java
+++ b/Mage.Tests/src/test/java/org/mage/test/cards/triggers/dies/ReturnOnlyFromGraveyardTest.java
@@ -61,6 +61,7 @@ public class ReturnOnlyFromGraveyardTest extends CardTestPlayerBase {
 
         castSpell(1, PhaseStep.POSTCOMBAT_MAIN, playerB, "Lightning Bolt", "Academy Rector");
 
+        setChoice(playerA, "When enchanted creature dies"); // Select triggered ability to execute last
         setChoice(playerA, "Yes");
         addTarget(playerA, "Primal Rage");
 
@@ -94,7 +95,8 @@ public class ReturnOnlyFromGraveyardTest extends CardTestPlayerBase {
 
         castSpell(1, PhaseStep.BEGIN_COMBAT, playerB, "Lightning Bolt", "Academy Rector");
 
-        setChoice(playerA, "Yes");
+        setChoice(playerA, "When enchanted creature dies"); // Select triggered ability to execute last
+        setChoice(playerA, "Yes"); // May exile it
         addTarget(playerA, "Primal Rage");
 
         setStopAt(1, PhaseStep.END_COMBAT);