From 44ee8e66a6c274b5fd3106f49782b7b110a2a314 Mon Sep 17 00:00:00 2001
From: Parnic <parnic@parnic.com>
Date: Tue, 20 Jul 2010 01:23:44 +0000
Subject: [PATCH] - fixed TargetTargetHealth module from not hiding when the
 player was the ToT and the self-display mode was set to "hide"

---
 modules/TargetOfTargetHealth.lua | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/modules/TargetOfTargetHealth.lua b/modules/TargetOfTargetHealth.lua
index 81dea9f..a700924 100644
--- a/modules/TargetOfTargetHealth.lua
+++ b/modules/TargetOfTargetHealth.lua
@@ -70,7 +70,7 @@ function TargetTargetHealth.prototype:GetOptions()
 			IceHUD.IceCore:SetColor("SelfColor", r, g, b)
 		end,
 		disabled = function()
-			return not self.moduleSettings.enabled or self.moduleSettings.selfDisplayMode ~= "Color as SelfColor"
+			return not self.moduleSettings.enabled
 		end,
 		hasAlpha = false,
 		order = 45,
@@ -83,14 +83,14 @@ end
 function TargetTargetHealth.prototype:Enable(core)
 	self.registerEvents = false
 	TargetTargetHealth.super.prototype.Enable(self, core)
-	
+
 	if self.moduleSettings.useSelfColor ~= nil then
 		if self.moduleSettings.useSelfColor == true then
 			self.moduleSettings.selfDisplayMode = "Color as SelfColor"
 		else
 			self.moduleSettings.selfDisplayMode = "Normal"
 		end
-		
+
 		self.moduleSettings.useSelfColor = nil
 	end
 
@@ -130,11 +130,19 @@ function TargetTargetHealth.prototype:Update(unit)
 		if self.moduleSettings.selfDisplayMode == "Color as SelfColor" then
 			self.color = "SelfColor"
 		elseif self.moduleSettings.selfDisplayMode == "Hide" then
+			if self:IsVisible() then
+				UnregisterUnitWatch(self.frame)
+			end
+			
 			self:Show(false)
 			return
 		end
 	end
-	
+
+	if not self:IsVisible() then
+		RegisterUnitWatch(self.frame)
+	end
+
 	self:Show(true)
 
 	self.determineColor = false