diff --git a/scripts/ai/tpl/generic.lua b/scripts/ai/tpl/generic.lua
index 4e29671..0bfdda2 100644
--- a/scripts/ai/tpl/generic.lua
+++ b/scripts/ai/tpl/generic.lua
@@ -33,9 +33,12 @@ function control()
 
   -- Think for attacking.
   elseif task == "attack" then
+    target = ai.target()
     -- Runaway if needed.
-    if(shield_run > 0 and ai.pshield() < shield_run) or
-        (armour_run > 0 and ai.parmour() < armour_run) then
+    if(shield_run > 0 and ai.pshield() < shield_run
+      and ai.pshield() < ai.pshield(target)) or
+        (armour_run > 0 and ai.parmour() < armour_run
+        and ai.parmour() < ai.parmour(target)) then
       ai.pushtask(0, "runaway", ai.target())
 
     -- Think like normal.