diff --git a/modtools/set-belief.lua b/modtools/set-belief.lua
index c46b2adaf..ae9f53062 100644
--- a/modtools/set-belief.lua
+++ b/modtools/set-belief.lua
@@ -153,7 +153,7 @@ end
 -- Gives a list of all the unit's beliefs and their values.
 -- Returns a table where the keys are the belief names and its values are its strength
 -- If tiers is true, the value is the tier of the belief instead of the trait's strength
-local function getUnitBeliefList(unit, tiers)
+function getUnitBeliefList(unit, tiers)
   local list = {}
 
   for id, beliefName in ipairs(df.value_type) do
diff --git a/modtools/set-personality.lua b/modtools/set-personality.lua
index 9b1b421a7..16428310e 100644
--- a/modtools/set-personality.lua
+++ b/modtools/set-personality.lua
@@ -85,7 +85,7 @@ function getUnitTraitCurrent(unit, trait)
   local base = getUnitTraitBase(unit, trait)
   local value = base
   if unit.status.current_soul.personality.temporary_trait_changes ~= nil then
-    value = value + unit.status.current_soul.personality.temporary_trait_changes[trait]
+    value = value + unit.status.current_soul.personality.temporary_trait_changes.traits[trait]
   end
 
   return value