From 36fe4b4b1d1018889900a3f255bf0149bca7e1ca Mon Sep 17 00:00:00 2001
From: Shadow <cata7007@gmail.com>
Date: Wed, 27 Nov 2013 11:21:20 +0200
Subject: [PATCH] Update Units.php

---
 GameEngine/Units.php | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/GameEngine/Units.php b/GameEngine/Units.php
index 84070181..92c9ca73 100644
--- a/GameEngine/Units.php
+++ b/GameEngine/Units.php
@@ -242,13 +242,14 @@ class Units {
 
     }
     
-    public function returnTroops($wref) {
+    public function returnTroops($wref,$mode=0) {
         global $database;
-
-        $getenforce=$database->getEnforceVillage($wref,0);
-        foreach($getenforce as $enforce) {
-            $this->processReturnTroops($enforce);    
-        }
+        if (!mode) {
+            $getenforce=$database->getEnforceVillage($wref,0);
+            foreach($getenforce as $enforce) {
+                $this->processReturnTroops($enforce);    
+            }
+        }    
         //check oasis
         $getenforce1=$database->getOasisEnforce($wref,1);
         foreach($getenforce1 as $enforce) {