From c095651eaa49c8ffcf7e008cd92610f79630c34a Mon Sep 17 00:00:00 2001
From: friarsol <zenchristo@gmail.com>
Date: Wed, 13 Jan 2021 21:20:41 -0500
Subject: [PATCH] Small functional improvement

---
 forge-ai/src/main/java/forge/ai/ability/PumpAi.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/forge-ai/src/main/java/forge/ai/ability/PumpAi.java b/forge-ai/src/main/java/forge/ai/ability/PumpAi.java
index b66018378ce..c7ee3ff7c13 100644
--- a/forge-ai/src/main/java/forge/ai/ability/PumpAi.java
+++ b/forge-ai/src/main/java/forge/ai/ability/PumpAi.java
@@ -76,7 +76,7 @@ public class PumpAi extends PumpAiBase {
             }
         } else if (aiLogic.equals("SwitchPT")) {
             // Some more AI would be even better, but this is a good start to prevent spamming
-            if (sa.isAbility() && sa.getActivationsThisTurn() > 0 && sa.getTargetRestrictions() == null) {
+            if (sa.isAbility() && sa.getActivationsThisTurn() > 0 && !sa.usesTargeting()) {
                 // Will prevent flipping back and forth
                 return false;
             }