diff --git a/src/adv/AdvBattle.lua b/src/adv/AdvBattle.lua
index dc5f201..5eaed83 100644
--- a/src/adv/AdvBattle.lua
+++ b/src/adv/AdvBattle.lua
@@ -1,4 +1,4 @@
-local Player, Enemy = table.unpack(require "adv.AdvPlayer")
+local Player, Enemy, Build = table.unpack(require "adv.AdvPlayer")
 local Buff = require "adv.AdvBuff"
 local Passive = require "adv.AdvPassive"
 local Battle = class("Battle")
diff --git a/src/adv/AdvPlayer.lua b/src/adv/AdvPlayer.lua
index 35f6011..15ccfa8 100644
--- a/src/adv/AdvPlayer.lua
+++ b/src/adv/AdvPlayer.lua
@@ -707,11 +707,11 @@ function Build:getTeam(nType, noSelf, mapIdx, includeLock)
 	end
 	return team
 end
-function Enemy:unlock()
+function Build:unlock()
 	self.lock = nil
 end
 
-function Enemy:getDB()
+function Build:getDB()
 	local db = {}
 	db.passives = {}
 	for _, passive in ipairs(self.passives) do
@@ -721,4 +721,4 @@ function Enemy:getDB()
 end
 
 
-return table.pack(Player, Enemy)
\ No newline at end of file
+return table.pack(Player, Enemy, Build)
\ No newline at end of file
--
libgit2 0.21.2