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;
             }