From 77b30de8f799532458392041dd174b05750b4977 Mon Sep 17 00:00:00 2001
From: KlazHTT <account@haothitran.com>
Date: Fri, 8 Mar 2024 23:24:26 -0500
Subject: [PATCH] Fix function to retrieve current loot specialization

---
 Core/Functions.lua | 15 +++++----------
 1 file changed, 5 insertions(+), 10 deletions(-)

diff --git a/Core/Functions.lua b/Core/Functions.lua
index f59d5d3..f9d17e5 100644
--- a/Core/Functions.lua
+++ b/Core/Functions.lua
@@ -31,19 +31,14 @@ end
 --------------------------------------------------------------------------------
 
 T.GetLootSpec = function()
-  local currentSpec = GetSpecialization()
-  local currentSpecName = select(2, GetSpecializationInfo(currentSpec))
   local lootSpecID = GetLootSpecialization()
-  local lootSpec = select(2, GetSpecializationInfoByID(lootSpecID))
+  local currentSpec = GetSpecialization()
+  local currentSpecName = currentSpec and select(2,GetSpecializationInfo(currentSpec)) or 'None'
 
-  -- if loot spec was manually set
-  if lootSpec then
-    return '|cnWHITE_FONT_COLOR:Loot:|r '..lootSpec..'    '
-  -- otherwise use current loot spec
-  elseif currentSpec then
-    return '|cnWHITE_FONT_COLOR:Loot:|r '..currentSpecName..'    '
+  if lootSpecID ~= 0 then
+    return '|cnWHITE_FONT_COLOR:Loot:|r '..select(2, GetSpecializationInfoByID(lootSpecID))..'    '
   else
-    return ''
+    return '|cnWHITE_FONT_COLOR:Loot:|r '..currentSpecName..'    '
   end
 end