what.

Putnam3145 · Apr 25, 2016 · d840f3c · d840f3c
1 parent 2a94a54
commit d840f3c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/raw/scripts/fortbent/extra_gui.lua b/raw/scripts/fortbent/extra_gui.lua
@@ -54,7 +54,7 @@ function getClaspect(unit)
     local currentClass = unitClasses['Current']
     local classes = persistTable.GlobalTable.roses.ClassTable
     local currentClassName = currentClass['Name']
-    if not currentClassLevel[currentClassName] then return {class=nil,color=nil} end
+    if not unitClasses[currentClassName] then return {class=nil,color=nil} end
     local currentClassLevel = tonumber(unitClasses[currentClassName]['Level'])+1
     local ofLocations={currentClassName:find('_OF_')}
     local aspectColor=sburbColors[currentClassName:sub(ofLocations[2]+1,-1)]
@@ -198,4 +198,4 @@ dfhack.onStateChange.fortbent_extra_gui=function(code)
         local viewfunc=viewscreenActions[dfhack.gui.getCurViewscreen()._type]
         if viewfunc then viewfunc() end
     end
-end
+end