From d840f3cd386aa23276cebdd7dedc4b70857a157d Mon Sep 17 00:00:00 2001
From: Putnam3145 <bpwhite@cableone.net>
Date: Sun, 24 Apr 2016 20:49:13 -0700
Subject: [PATCH] what.

---
 raw/scripts/fortbent/extra_gui.lua | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/raw/scripts/fortbent/extra_gui.lua b/raw/scripts/fortbent/extra_gui.lua
index 7f27319..eeb0243 100644
--- 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
\ No newline at end of file
+end