use library function for getting the caste raw

DFHack · Sep 8, 2024 · 0471d9e · 0471d9e
1 parent a6cdf8c
commit 0471d9e
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/rejuvenate.lua b/rejuvenate.lua
@@ -7,12 +7,9 @@ local DEFAULT_OLD_AGE = 160
 local ANY_BABY = df.global.world.units.other.ANY_BABY
 
 local function get_caste_misc(unit)
-    local cre = df.creature_raw.find(unit.race)
-    if not cre then return end
-    if unit.caste < 0 or unit.caste >= #cre.caste then
-        return
-    end
-    return cre.caste[unit.caste].misc
+    local craw = dfhack.units.getCasteRaw(unit)
+    if not craw then return end
+    return craw.misc
 end
 
 local function get_adult_age(misc)