diff --git a/ai.cpp b/ai.cpp
index 7bb6397..2a00f12 100644
--- a/ai.cpp
+++ b/ai.cpp
@@ -167,8 +167,8 @@ namespace ai {
     if(active()) {
       std::sort(plan.script.begin(), plan.script.end(),
           [&](auto& l, auto& r) {
-            int l_cost = l.cost + (!l.can_effect(start) * ai::SCORE_MAX);
-            int r_cost = r.cost + (!r.can_effect(start) * ai::SCORE_MAX);
+            int l_cost = l.cost + ai::distance_to_goal(start, goal);
+            int r_cost = r.cost + ai::distance_to_goal(start, goal);
             return l_cost < r_cost;
           });
     }