From 225fd465c1156a4cbe200b4417bc1ad91ce3ccfc Mon Sep 17 00:00:00 2001
From: ZomgPonies <lorden@zomgponies.com>
Date: Wed, 28 May 2014 14:36:55 -0400
Subject: [PATCH] Post Test #2 bugfixes

---
 code/game/gamemodes/nations/flagprocs.dm | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/code/game/gamemodes/nations/flagprocs.dm b/code/game/gamemodes/nations/flagprocs.dm
index 161831183ad..e058ab41fb3 100644
--- a/code/game/gamemodes/nations/flagprocs.dm
+++ b/code/game/gamemodes/nations/flagprocs.dm
@@ -36,8 +36,12 @@
 /obj/item/flag/nation/attack_hand(mob/user as mob)
 	if(user.mind)
 		if(user.mind.nation)
+			var/obj/item/flag/nation/F = locate(user.mind.nation.flagpath)
 			if(user.mind.nation.type == nation) //Same team as flag
-				if(loc != startloc)
+				if(liege && liege == F.liege)
+					user << "<span class='warning'>You can't steal your liege's flags!</span>"
+					return
+				else if(loc != startloc)
 					loc = startloc
 					user.visible_message("[user] sends [src] back to base!", "You return [src] to your base!")
 					return
@@ -45,13 +49,9 @@
 					user << "<span class='warning'>You can't move your flag from it's home location!</span>"
 					return
 			else
-				var/obj/item/flag/nation/F = locate(user.mind.nation.flagpath)
 				if(captured && Adjacent(F.startloc))
 					user << "<span class='warning'>This flag has already been captured by your team!</span>"
 					return
-				if(liege && liege == F.liege)
-					user << "<span class='warning'>You can't steal your liege's flags!</span>"
-					return
 				else if(..())
 					captured = 0
 					anchored = 0
@@ -82,17 +82,17 @@
 			//Announce capture/vassalage here.
 			for(var/mob/living/carbon/human/H in player_list)
 				if(H.mind && H.mind.nation)
-					if(H.mind.nation.type == F.nation)
+					if(H.mind.nation == F.nation)
 						H << "<span class='warning'>You have just vassalized [nation.name]! They must now obey any memebrs of your nation!</span>"
 						continue
-					else if(H.mind.nation.type == nation)
+					else if(H.mind.nation == nation)
 						H << "<span class='warning'>You are now vassals of [liege.name]! You must now obey the orders of any of their members!</span>"
 						continue
 			//Check for Victory
 			for(var/obj/item/flag/nation/N in flag_list)
 				if(N == F)
 					continue
-				if(N.captured && (N.liege == F.nation) && N.Adjacent(F))
+				if(N.captured && (N.liege == F.nation) && N.Adjacent(F.startloc))
 					continue
 				else
 					return