Reverting more changes

rgd87 · Dec 9, 2020 · acc1148 · acc1148
1 parent e275f70
commit acc1148
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/NugEnergy.lua b/NugEnergy.lua
@@ -531,6 +531,7 @@ function NugEnergy.Initialize(self)
             PowerFilter = nil
             PowerTypeIndex = nil
             self:UnregisterEvent("UNIT_POWER_UPDATE")
+            self:UnregisterEvent("UNIT_POWER_FREQUENT")
             self:UnregisterEvent("UNIT_MAXPOWER")
             self:UnregisterEvent("PLAYER_REGEN_DISABLED")
             self:SetScript("OnUpdate", nil)
@@ -560,11 +561,14 @@ function NugEnergy.Initialize(self)
                 if isClassic and NugEnergyDB.enableClassicTicker then
                     GetPower = GetPower_ClassicRogueTicker(nil, 19, 0, false)
                     NugEnergy.UNIT_MAXPOWER = UNIT_MAXPOWER_ClassicTicker
+                    self:UnregisterEvent("UNIT_POWER_FREQUENT")
                     self:SetScript("OnUpdate",self.UpdateEnergy)
                     ClassicTickerFrame:Enable()
                     self:UpdateBarEffects()
                 else
                     GetPower = RageBarGetPower(nil, 5, nil, true)
+                    self:RegisterUnitEvent("UNIT_POWER_FREQUENT", "player")
+                    self:SetScript("OnUpdate", nil)
                     if ClassicTickerFrame.isEnabled then
                         ClassicTickerFrame:Disable()
                         self:UpdateBarEffects()
@@ -577,7 +581,6 @@ function NugEnergy.Initialize(self)
                 self:UNIT_MAXPOWER()
                 self:RegisterEvent("PLAYER_REGEN_DISABLED")
                 self:UPDATE_STEALTH()
-                self:RegisterUnitEvent("UNIT_POWER_FREQUENT", "player")
             elseif newPowerType =="RAGE" and NugEnergyDB.rage then
                 PowerFilter = "RAGE"
                 PowerTypeIndex = Enum.PowerType.Rage
@@ -592,6 +595,7 @@ function NugEnergy.Initialize(self)
                 end
                 GetPower = RageBarGetPower(30, 10, nil, nil)
                 self:RegisterEvent("PLAYER_REGEN_DISABLED")
+                self:SetScript("OnUpdate", nil)
                 self:UNIT_MAXPOWER()
                 self:UPDATE_STEALTH()
             elseif newPowerType =="MANA" and isClassic then