diff --git a/modules/FocusHealth.lua b/modules/FocusHealth.lua
index 11e4816..7f71da2 100644
--- a/modules/FocusHealth.lua
+++ b/modules/FocusHealth.lua
@@ -267,6 +267,10 @@ function FocusHealth.prototype:Enable(core)
 	self:CreateRaidIconFrame()
 
 	self:Update(self.unit)
+
+	-- for showing/hiding the frame based on unit visibility
+	self.frame:SetAttribute("unit", self.unit)
+	RegisterUnitWatch(self.frame)
 end
 
 function FocusHealth.prototype:CreateBackground()