Fixed retreats calculation.

jwhitmire · Jul 4, 2013 · 7cf83ca · 7cf83ca
1 parent f93e2d5
commit 7cf83ca
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/adjudicator/state.rb b/lib/adjudicator/state.rb
@@ -70,7 +70,9 @@ def set_area_unit(area, unit)
     def apply_orders!(orders)
       orders.each do |order|
         if Move === order && order.succeeded?
-          @retreats[order.dst] = area_unit(order.dst)
+          if (dislodged_unit = area_unit(order.dst))
+            @retreats[order.dst] = area_unit(order.dst)
+          end
 
           set_area_unit(order.dst, area_unit(order.unit_area))
           set_area_unit(order.unit_area, nil)